- Install dependencies:
pip install -r requirements.txt
Train and Test
- Put your data into the 'datasets' directory
- Creat train and val patches:
python get_patch_wdc.py
- Creat test data:
python get_patch_wdc_test.py
- Open the model directory in terminal
- Set python environment
- Train example:
python .\train.py --config-file .\gauss_blind_config.yaml
- Test example:
python .\predict.py --config-file .\gauss_blind_config.yaml