Merge branch 'main' of https://github.com/AarohiSingla/SRGAN_CustomDa…

…taset
AarohiSingla · Dec 18, 2022 · c0227b4 · c0227b4
2 parents 4b356eb + 05783b0
commit c0227b4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -16,3 +16,15 @@ conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
 
 ##### CPU Only
 conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch
+
+
+#### Train your Model:
+!python main.py --LR_path custom_dataset_cars/hr_train_LR --GT_path custom_dataset_cars/hr_train_HR
+
+#### Test your Model:
+!python main.py --mode test_only --LR_path test_data/cars --generator_path ./model/srgan_custom.pt
+
+
+#### If you want to test the model which is trained on div2K dataset then download the model from here:  https://github.com/AarohiSingla/SRGAN_CustomDataset/tree/main/pretrained_models
+
+