I designed a model in Pytorch that performs image translation from the input images to the target images. An example of the input image and its target image is shown below
Run train.py
script.
Dependencies are :
Python 3.10
Numpy 1.23.5
Torch 1.12.1
Torchvison 0.13.1
Cuda 11.6
Matplotlib 3.6.2
Pillow 9.3.0
Download the model from here. Model's name is mymodel_version3.pth
Run test_oneimage.py
on a test image
Dataset for training and testing is provided by Visidon