/CORN202

Clothing image classifier using a recurrent neural network. Winner of the MAIS202 AI contest.

Primary LanguagePureBasic

CORN202

Final project for the MAIS202 bootcamp, about classifying images of clothing items using machine learning.

How to run the project:

1- The train_images.npy file (https://www.kaggle.com/competitions/mais-202-winter-2022/data) has to be added to the dataset folder

2- Run model.py to train a new model

3- Run predict.py with the path of where the model is saved (line 43) to generate predictions

Dependencies:

  • Numpy
  • Pandas
  • Tensorflow
  • Matplotlib
  • Keras
  • seaborn
  • sklearn