Study of synthesis images noise detection using 26 attributes
Jérôme BUISINE a5e97a807e Merge branch 'release/v0.1.9' | 5 gadi atpakaļ | |
---|---|---|
analysis | 5 gadi atpakaļ | |
data_processing | 5 gadi atpakaļ | |
display | 5 gadi atpakaļ | |
generate | 5 gadi atpakaļ | |
modules @ 2957359371 | 5 gadi atpakaļ | |
optimization @ 6dcd32014a | 5 gadi atpakaļ | |
others | 5 gadi atpakaļ | |
prediction | 5 gadi atpakaļ | |
run | 5 gadi atpakaļ | |
simulation | 5 gadi atpakaļ | |
.gitignore | 5 gadi atpakaļ | |
.gitmodules | 5 gadi atpakaļ | |
LICENSE | 5 gadi atpakaļ | |
README.md | 5 gadi atpakaļ | |
custom_config.py | 5 gadi atpakaļ | |
data_attributes.py | 5 gadi atpakaļ | |
find_best_attributes.py | 5 gadi atpakaļ | |
find_best_filters.py | 5 gadi atpakaļ | |
models.py | 5 gadi atpakaļ | |
requirements.txt | 5 gadi atpakaļ | |
train_model.py | 5 gadi atpakaļ | |
train_model_attributes.py | 5 gadi atpakaļ | |
train_model_filters.py | 5 gadi atpakaļ |
Noise detection on synthesis images with 26 attributes obtained using few filters.
Filters list:
pip install -r requirements.txt
Generate all needed data for each features (which requires the whole dataset. In order to get it, you need to contact us).
python generate/generate_all_data.py --feature all
You have to create a symbolic link to your own database which respects this structure:
Create your symbolic link:
ln -s /path/to/your/data dataset
modules/config/global_config.py
run/runAll_*.sh
script.model_comparisons.csv
file used for store models performance.Remark: Note here that all python script have --help command.
python generate_data_model.py --help
Parameters explained:
data
folder.This is an example of how to train a model
python train_model.py --data 'data/xxxx' --output 'model_file_to_save' --choice 'model_choice'
Expected values for the choice parameter are ['svm_model', 'ensemble_model', 'ensemble_model_v2'].
Now we have a model trained, we can use it with an image as input:
python prediction/predict_noisy_image_svd.py --image path/to/image.png --interval "x,x" --model saved_models/xxxxxx.joblib --feature 'lab' --mode 'svdn' --custom 'min_max_filename'
The model will return only 0 or 1:
All SVD features developed need:
custom_config.py
file.data_attributes.py
file.Now we have a model trained, we can use it with an image as input:
python prediction_scene.py --data path/to/xxxx.csv --model saved_model/xxxx.joblib --output xxxxx --scene xxxx
Remark: scene parameter expected need to be the correct name of the Scene.
All scripts with names display/display_*.py are used to display data information or results.
Just use --help option to get more information.
All scripts named prediction/predict_seuil_expe*.py are used to simulate model prediction during rendering process. Do not forget the custom parameter filename if necessary.
Once you have simulation done. Checkout your threshold_map/%MODEL_NAME%/simulation_curves_zones_*/ folder and use it with help of display_simulation_curves.py script.