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[2]
w = data.shape[3]
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.