FashionMnist repository contains code to classify the fashion-mnist dataset through kaggle and
their dataset. NOTE: This means we do not have the labels for the test set, which makes this project a little different than
most fashion mnist projects, since they can test their model much easier, we get 5 submissions per day and the accuracy result
is only from 30% of the actual test data. Right now the training is being done in python using keras and a tensorflow
backend. Plans to experiment with Matlab’s gpu computing toolkit are in the future. See below for more information on the project.
Here are some things you’ll need to get started.
- numpy:
pip3 install numpy - tensorflow with gpu support:
pip3 install tensorflow-gpu - keras:
pip3 install keras - python imaging library:
pip3 install Pillow - Process and system utilities library
pip3 install psutil - Kaggle API library
pip3 install kaggle
The csv files are too large for us to put in the repository.
- Create a folder called Data
- Download the dataset here
- Put train.csv and test.csv in the Data folder you created.
Here are some of the descriptions for each of our classes.
This class is used to read in all the files necessary for the training including the training images, testing images, and CSV files, which are the tasks handled by each function from within the class. Once the training has been completed, this class generates the CSV file containing the resulting data from the training session.
The functions for processing the images utilize matrix operations to generate the new training sets through normalization, shifting flipping, and shuffling the images, and adding contrast to the given data sets.
Used to save data augmentation and analyze the data.
Holds the code to train, load, and build networks.
Leave a Reply