update readme

main
protsenkovi 7 months ago
parent 0dd154d0cd
commit ac3c9cce04

@ -2,12 +2,12 @@
Example Example
``` ```
python train.py --model SRNetRot90 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 transfer_to_lut.py
python train.py --model_path /wd/lut_reproduce/models/last_transfered_lut.pth --total_iter 2000 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 python image_demo.py
``` ```
@ -15,13 +15,17 @@ python image_demo.py
Help Help
``` ```
python train.py --help python train.py --help
python validate.py --help python test.py --help
python transfer_to_lut.py --help python transfer_to_lut.py --help
python image_demo.py --help python image_demo.py --help
``` ```
Requirements: Requirements:
```
pip install shedulefree tensorboard opencv-python-headless scipy pandas
```
- [shedulefree](https://github.com/facebookresearch/schedule_free) - [shedulefree](https://github.com/facebookresearch/schedule_free)
- tensorboard - tensorboard
- opencv-python-headless - opencv-python-headless
- scipy - scipy
- pandas
Loading…
Cancel
Save