01. pytorch 为什么 plt.imshow(np.transpose(npimg, (1, 2, 0)))
解释这句话:plt.imshow(np.transpose(npimg, (1, 2, 0)))。因为在plt.imshow在现实的时候输入的是
(imagesize,imagesize,channels),而def imshow(img,text,should_save=False)中,参数img的格式为
(channels,imagesize,imagesize),这两者的格式不一致,我们需要调用一次np.transpose函数,
即np.transpose(npimg,(1,2,0)),将npimg的数据格式由(channels,imagesize,imagesize)转化为
(imagesize,imagesize,channels)
来源:CSDN
作者:snow_lee_liwei
链接:https://blog.csdn.net/snow_lee_liwei/article/details/103639387