|
|
|
@ -2,12 +2,12 @@
|
|
|
|
|
Example
|
|
|
|
|
```
|
|
|
|
|
python train.py --model SRNetRot90
|
|
|
|
|
python validate.py --val_datasets Set5,Set14,B100,Urban100,Manga109
|
|
|
|
|
python test.py --test_datasets Set5,Set14,B100,Urban100,Manga109
|
|
|
|
|
|
|
|
|
|
python transfer_to_lut.py
|
|
|
|
|
|
|
|
|
|
python train.py --model_path /wd/lut_reproduce/models/last_transfered_lut.pth --total_iter 2000
|
|
|
|
|
python validate.py --val_datasets Set5,Set14,B100,Urban100,Manga109
|
|
|
|
|
python test.py --test_datasets Set5,Set14,B100,Urban100,Manga109
|
|
|
|
|
|
|
|
|
|
python image_demo.py
|
|
|
|
|
```
|
|
|
|
@ -15,13 +15,17 @@ python image_demo.py
|
|
|
|
|
Help
|
|
|
|
|
```
|
|
|
|
|
python train.py --help
|
|
|
|
|
python validate.py --help
|
|
|
|
|
python test.py --help
|
|
|
|
|
python transfer_to_lut.py --help
|
|
|
|
|
python image_demo.py --help
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
Requirements:
|
|
|
|
|
```
|
|
|
|
|
pip install shedulefree tensorboard opencv-python-headless scipy pandas
|
|
|
|
|
```
|
|
|
|
|
- [shedulefree](https://github.com/facebookresearch/schedule_free)
|
|
|
|
|
- tensorboard
|
|
|
|
|
- opencv-python-headless
|
|
|
|
|
- scipy
|
|
|
|
|
- pandas
|