Syntesis images noise detection using CNN approach

jbuisine 5331476f2b Merge branch 'release/v0.0.2' il y a 6 ans
img_train a22b32b71a First CNN model version il y a 6 ans
img_validation a22b32b71a First CNN model version il y a 6 ans
models a22b32b71a First CNN model version il y a 6 ans
.gitignore a22b32b71a First CNN model version il y a 6 ans
README.md db0864c00b Update of README file il y a 6 ans
TODO.md a22b32b71a First CNN model version il y a 6 ans
classification_cnn_keras.py a22b32b71a First CNN model version il y a 6 ans
generate_dataset.py 44d28a5898 Update of documentation and requirements il y a 6 ans
requirements.txt 44d28a5898 Update of documentation and requirements il y a 6 ans

README.md

Noise detection project

Requirements

pip install -r requirements.txt

How to use

Generate dataset (run only once time) :

python generate_dataset.py

It will split scenes and generate all data you need for your neural network. You can specify the number of sub images you want in the script by modifying NUMBER_SUB_IMAGES variables.

After your built your neural network in classification_cnn_keras.py, you just have to run it :

python classification_cnn_keras.py

How to contribute

This git project uses git-flow implementation. You are free to contribute to it.