Skip to content

abhigoyal1997/transformed-mnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CNNs on Transformed MNIST

Analysis of different CNN models on a randomly scaled and translated MNIST dataset using a multi-label setup (for generalisation in case of multi-digit).

Some Results

These models have been trained on 10000 images from official training split of MNIST after random scaling and translation using a Multi-Label-Soft-Margin loss. The results are reported as average F1-score of prediction on the official 10000 test images from MNIST after random scale and translation.

  1. 2 Convolution layers followed by 3 Dense layers: ~0.732
  2. A model similar to AG-CNN:
    • Global branch (the model mentioned above): ~0.732
    • Local branch using localized images: ~0.938
    • Fused Global and Local branch: ~0.957

References

  1. AG-CNN: Diagnose like a radiologist: Attention guided convolutional neural network for thorax disease classification
  2. MNIST Dataset

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published