Pytorch_trick_01

∥☆過路亽.° 提交于 2019-12-28 05:29:23

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)
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!