Skip to content

DenseNet Neural Network used for image classification on CIFAR-10 dataset

Notifications You must be signed in to change notification settings

yashnaik99/CNN-on-CIFAR-10

Repository files navigation

CNN-on-CIFAR-10

DenseNet NN used for image classification on CIFAR-10 dataset

In this repository I have used a DenseNet architecture to classify images in the CIFAR-10 dataset.

The assignment instructions are as follows:

  1. Please refer to the DenseNet pdf file link to access the state-of-art DenseNet code for reference - DenseNet - cifar10 notebook link
  2. You need to create a copy of this and "retrain" this model to achieve 90+ test accuracy.
  3. You cannot use DropOut layers.
  4. You MUST use Image Augmentation Techniques.
  5. You cannot use an already trained model as a beginning points, you have to initilize as your own
  6. You cannot run the program for more than 300 Epochs, and it should be clear from your log, that you have only used 300 Epochs
  7. You cannot use test images for training the model.
  8. You cannot change the general architecture of DenseNet (which means you must use Dense Block, Transition and Output blocks as mentioned in the code)
  9. You are free to change Convolution types (e.g. from 3x3 normal convolution to Depthwise Separable, etc)
  10. You cannot have more than 1 Million parameters in total
  11. You are free to move the code from Keras to Tensorflow, Pytorch, MXNET etc.
  12. You can use any optimization algorithm you need.
  13. You can checkpoint your model and retrain the model from that checkpoint so that no need of training the model from first if you lost at any epoch while training. You can directly load that model and Train from that epoch.

About

DenseNet Neural Network used for image classification on CIFAR-10 dataset

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published