My Code is listed below:
def train(epoch): model.train() for batch_idx, (data, target) in enumerate(trainloader): #print str(target.numpy()) h = data.shape w = data.shape c = 3 output = data.squeeze(0) output = output.numpy() output = np.resize(output,(h,w,c)) plt.imshow(output) plt.show() exit(0)
Actually, I want to get a output whose locations of each pixels are as same as my original input image, and however my final output is very different from my input image. So, how can i solve this problem?
Thank you for your help.