Skip to content
Permalink
8d39349202
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?

ECE50024_Project/README.md

Go to file
 
 
Cannot retrieve contributors at this time
Requirements Training Testing
37 lines (20 sloc) 416 Bytes
Raw Blame

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