Study of synthesis images noise detection using 26 attributes

Jérôme BUISINE 285848c290 Merge branch 'release/v0.2.8' 4 年之前
analysis cb6026f2c7 Add of 26 features metric 5 年之前
generate c409191b1c update prediction script for new dataset structure 4 年之前
modules @ cebf2adbf1 0a05939b74 Update of optimization process using backups 5 年之前
optimization @ db3538337f 0a05939b74 Update of optimization process using backups 5 年之前
prediction c409191b1c update prediction script for new dataset structure 4 年之前
.gitignore 0a05939b74 Update of optimization process using backups 5 年之前
.gitmodules b693a667bb Add of optimization modules 5 年之前
LICENSE dc0463b6b5 Project initialization 5 年之前
README.md 30d257e0f7 update of the whole project to enable use of new dataset 4 年之前
custom_config.py e567b11e90 add svm gpu usage 4 年之前
data_attributes.py 788e1c045a add of svd entropy as new complexity measure 4 年之前
find_best_attributes.py b73b27ab44 add balanced data into SVC 4 年之前
find_best_filters.py b73b27ab44 add balanced data into SVC 4 年之前
models.py 55b2331fb9 model score update 4 年之前
requirements.txt c409191b1c update prediction script for new dataset structure 4 年之前
train_model.py c409191b1c update prediction script for new dataset structure 4 年之前
train_model_attributes.py b73b27ab44 add balanced data into SVC 4 年之前
train_model_filters.py b73b27ab44 add balanced data into SVC 4 年之前

README.md

Noise detection using 26 attributes

Description

Noise detection on synthesis images with 26 attributes obtained using few filters.

Filters list:

  • average
  • wiener
  • median
  • gaussian
  • wavelet

Requirements

pip install -r requirements.txt

Project structure

Link to your dataset

You need database which respects this structure:

  • dataset/
    • Scene1/
    • Scene1_00050.png
    • Scene1_00070.png
    • ...
    • Scene1_01180.png
    • Scene1_01200.png
    • Scene2/
    • ...
    • ...

Code architecture description

  • modules/*: contains all modules usefull for the whole project (such as configuration variables)
  • analysis/*: contains all jupyter notebook used for analysis during thesis
  • generate/*: contains python scripts for generate data from scenes (described later)
  • data_processing/*: all python scripts for generate custom dataset for models
  • prediction/*: all python scripts for predict new threshold from computed models
  • data_attributes.py: files which contains all extracted features implementation from an image.
  • custom_config.py: override the main configuration project of modules/config/global_config.py
  • train_model.py: script which is used to run specific model available.

Generated data directories:

  • data/*: folder which will contain all generated .train & .test files in order to train model.
  • data/saved_models/*: all scikit learn or keras models saved.
  • data/models_info/*: all markdown files generated to get quick information about model performance and prediction obtained after running run/runAll_*.sh script.
  • data/results/: This folder contains model_comparisons.csv file used for store models performance.

License

The MIT license