numpy类型image
e.g.: image.shape是[1080,1920,3]
import matplotlib.pyplot as plt
plt.imshow(image)
tensor类型image
e.g.: image.shape是[320,320,3] / [320,3,320] / [320,3,320]
import torchvision.transforms as transformers
import matplotlib.pyplot as plt
unloader = transformers.ToPILImage()
image = unloader(image.squeeze())
plt.imshow(image)