Syntesis images noise detection using CNN approach
|
il y a 3 ans | |
---|---|---|
analysis | il y a 3 ans | |
display | il y a 4 ans | |
generate | il y a 3 ans | |
models | il y a 4 ans | |
prediction | il y a 3 ans | |
run | il y a 3 ans | |
simulation | il y a 4 ans | |
.gitignore | il y a 4 ans | |
.gitmodules | il y a 4 ans | |
LICENSE | il y a 4 ans | |
README.md | il y a 3 ans | |
__init__.py | il y a 4 ans | |
cnn_models.py | il y a 3 ans | |
config.py | il y a 3 ans | |
prediction_model.py | il y a 4 ans | |
requirements.txt | il y a 4 ans | |
train_lstm_weighted.py | il y a 3 ans | |
train_model.py | il y a 3 ans |
git clone --recursive https://github.com/prise-3d/Thesis-NoiseDetection-CNN.git
pip install -r requirements.txt
modules/config/global_config.py
run/runAll_*.sh
script.model_comparisons.csv
file used for store models performance.Generate reconstructed data from specific method of reconstruction (run only once time or clean data folder before):
python generate/generate_reconstructed_data.py -h
Generate custom dataset from one reconstructed method or multiples (implemented later)
python generate/generate_dataset.py -h
List of expected parameter by reconstruction method:
Example:
python generate/generate_dataset_sequence_file.py --output data/output_data_filename --folder <generated_data_folder> --features "svd_reconstruction, ipca_reconstruction, fast_ica_reconstruction" --params "100, 200 :: 50, 10 :: 50" --sequence 5 --size "100, 100" --selected_zones <zones_files.csv>
Then, train model using your custom dataset:
python train_lstm_model.py --train data/custom_dataset.train --test data/custom_dataset.test --chanels "1,3,3" --epochs 30 --batch_size 64 --seq_norm 1 --output output_model_name
Now we have a model trained, we can use it with an image as input:
python prediction/predict_noisy_image.py --image path/to/image.png --model saved_models/xxxxxx.json --features 'svd_reconstruction' --params '100, 200'
The model will return only 0 or 1:
All scripts named prediction/predict_seuil_expe*.py are used to simulate model prediction during rendering process.
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.