hahazhar

``` def my_test(model, opt): dataset_dir = opt['name'] out_dir = os.path.join( "..", "experiments", dataset_dir) if not os.path.exists(out_dir): os.makedirs(out_dir) out_dir = os.path.join(out_dir, 'my_test') if not os.path.exists(out_dir): os.mkdir(out_dir) # load info filePath =...

Essentially, the image that is passed in, should be normalised. Pixel values should fall between 0 & 1 Afterwhich, it works well. One thing to note would be the size...

Have to add this line # process data Nimg = Nimg / 255.0 # Normalisation Nimg = torch.from_numpy(np.transpose(Nimg, (2, 0, 1))).type(torch.FloatTensor) Hope it works for you

Is it python 3.10 or python 3.9 ? there are a few dependency conflicts