-
Clone project
-
Install packages. They are all listed in requirements.txt. Most newer editors has an auto install feature.
-
Extract images in the images directory.
-
Run CNN.py. You can decide how many epochs on line 54 of the file. This will create an output file called model.h5. This is a full trained neural network model.
-
Test the model by running the file TestCNN.py. Adjust the image path on line 10, in order to use a test image of your choice.
jaxels10/CategoricalCNN
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|