ondrejzapletal/cnn_ilsvrc_tools

Multiple tools developed for masters thesis dealing with solution of ILSVRC with CNN.

Howto

Install the requirements

pip install -r requirements.txt

Create following file structure:

├── datasets
├── images
├── logs
├── models
├── tools
│   ├── application.py
│   ├── dataset_prepper.py
│   ├── evaluator.py
│   ├── extractor.py
│   ├── loggers.py
│   ├── model_creator.py
│   ├── requirements.txt
└── trained_models

Use this script to create dataset from images in images/ folder

python dataset_prepper.py

Use this script that creates model in models/ folder

python model_creator.py

Use this script to start training models on dataset in datasets/

python application.py