Requirements
- pytorch >=2.0
- torchvision
- os
- time
- PIL
Training
Select appropriate parameters in training.py
To train:
python3 training.py
Models will be saved to the 'models' folder
Testing
In testing.py change model_path to the path of the saved model, and change the test_img_path as required
To test:
python3 testing.py
Image will be saved to the 'outputs' folder