MyNeuralNetwork

This is my diploma project - convolution neural network for classification of airborne objects.

Project structure:

1. train.py - main file for training model
2. config.py - just configuration file
3. model.py - architecture of NN is here
4. test.py - contains function test_model to evaluate your model with test images
5. requirements.txt - all libs which you need to work with this project

Quick start

1. Choose directory on your pc and clone repository there:

   git clone https://github.com/Plinx28/MyNeuralNetwork.git;
2. Load your data to directory data/ (check file config.py to structure classes to train/valid/test samples);
3. Enter all parameters of images in the config.py;
4. Change something inside the train.py (if you want);
5. Just run train.py
6. Test your model with test.py