对 np.transpose() 的理解
np.transpose(x, (1, 2, 0))按照指定的axis顺序对array x 进行索引,若x中某个元素的index是(x1, x2, x3),则经上面变换以后其index变为(x2, x3, x0)。易知,np.transpose(x, (0, 1, 2))和 x 相同。在pytorch中图片的像素点是按照(c, H, W)排列的,我们可以利用上述变换将像素点按照(H, W, c)排布,从而使plt.imshow()可以读取。...
复制链接