You can run this command to begin the training.
python main.py --epoch=5000 --batch_size=128
You can put your images to demo/images directory and run this command to see the results:
python demo.py --scale=2
You can modify --scale argument to your desired value, it affects to your output image size
new_size = old_size * scale
- Python 3.8 or 3.9
- Tensorflow 2.5.0
- Numpy 1.19.1
- Matplotlib 3.4.3
- Pandas 1.3.4
- OpenCV 4.5.3