coding-hour-classification

There are several minor improvements made to the original files.

Quick overview

First of all you have to activate conda environment specified in environment.yml file. Some packages might be missing, so install on your own!
Then you can generate your own images (or use already created ones in created_images directory): python image_generator.py
Then you have to reorder them: python prepare_images.py
Then train your neural network: python train.py
Finally, check out your result: python image_predictor.py

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
created_images		created_images
sorted_images		sorted_images
README.md		README.md
environment.yml		environment.yml
image_generator.py		image_generator.py
image_predictor.py		image_predictor.py
model.pth		model.pth
model.py		model.py
painter.py		painter.py
prepare_images.py		prepare_images.py
train.py		train.py