diff --git a/.gitignore b/.gitignore index 2323b12..a0998bf 100644 --- a/.gitignore +++ b/.gitignore @@ -160,4 +160,6 @@ cython_debug/ #.idea/ *.png -*.0 \ No newline at end of file +*.0 + +*.txt \ No newline at end of file diff --git a/readme.md b/readme.md index 7b31dd7..6743cf6 100644 --- a/readme.md +++ b/readme.md @@ -1,7 +1,7 @@ Example ``` -python train.py --model SRNetR90 +python train.py --model SRNet python test.py --test_datasets Set5,Set14,B100,Urban100,Manga109 python transfer_to_lut.py @@ -11,23 +11,10 @@ python test.py --test_datasets Set5,Set14,B100,Urban100,Manga109 python image_demo.py ``` +For help use `--help`. -Help -``` -python train.py --help -python test.py --help -python transfer_to_lut.py --help -python image_demo.py --help -``` Requirements: ``` pip install schedulefree tensorboard opencv-python-headless scipy pandas matplotlib -``` - - -python train.py --model ChebyKANNet --total_iter 200000 --train_datasets mix-01-blur-1+ffmpeg --test_datasets mix-01-blur-test --upscale_factor 1 --val_step 210000 --layers_count 4 -python transfer_to_lut.py --model_path ../experiments/ChebyKANNet_RGB_mix-01-blur-1+ffmpeg_x1/checkpoints/ChebyKANNet_200000.pth -python test.py --model_path ../experiments/last_transfered_lut.pth --test_datasets src --save_predictions - -python test.py --model_path ../--test_datasets src --save_predictions \ No newline at end of file +``` \ No newline at end of file