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

 
 
 

Contributors