https://github.com/animesh-chourey/fashion-mnist-classification

https://github.com/animesh-chourey/fashion-mnist-classification

cnn fashion-mnist image-classification pytorch

Last synced: 2 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/animesh-chourey/fashion-mnist-classification
• Owner: Animesh-Chourey
• Created: 2022-08-28T16:31:19.000Z (about 2 years ago)
• Default Branch: master
• Last Pushed: 2022-08-28T17:47:24.000Z (about 2 years ago)
• Last Synced: 2024-08-05T12:56:55.689Z (3 months ago)
• Topics: cnn, fashion-mnist, image-classification, pytorch
• Language: Jupyter Notebook
• Homepage:
• Size: 297 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
Using PyTorch, on FashionMNIST dataset a convolutional neural network is trained for image classification task.

The dataset consists of 60,000 training images and 10,000 test images of grayscale nature. Each image is of 28x28 dimension and each is assisiated with a label from 10 classes (0-9).

The network's architecture implemented to execute the multi-class classification task is as following :

![](Architecture.png)

For training purpose the following conditions are set:

* ReLU as activation function

* Weights are initialised using Xavier initialisation

* SGD optimiser with learning rate of 0.1

* 50 epochs are used for training

Learning rate is changed to 0.001, 0.1, 0.5, 1, 10 with ReLU as activtion function. Later, different activation functions are used such as Tanh, Sigmoid and ELU to note their impact while training the model. Finally dropout is also used on the second fully connected layer to address overfitting.