CDWatson:

Note: This isn’t the same image as above, the thresholded images do correspond with their original counterpart and its not the images being edited completely.


I’m not sure I understand this statement completely.
Are all images visualized in an unexpected way or only a subset?
I don’t know what the range of the pixel values in the original image was, but note that ToTensor normalizes the data values to the range [0, 1].
Could you check, if matplotlib is using a wrong format/colormap and/or if the values in the transformed image look expected?