I’m not facing some error but just make sure this is the right way to put input image.
afterl I collect my data I try to visualize the image like
imgA = cracks_guess[0]

imgA = imgA.detach().cpu()

imgA = imgA.squeeze(0)

print(imgA.shape)

imgA = imgA.permute(1,2,0)

plt.imshow(imgA)

#plt.show()

but can not see the image