Well for that you apply the np.argmax function on the output of the network you will get an output in the same format as that of your labels and you can use opencv or pil to show that image after you have converted the cuda tensor to a numpy array on the cpu