由于读取到的图像是W*H*C 的格式，但是在输入网络的时候要求是C * W * H，因此在输入网络之前需要变换通道，通常采用transpose函数实现：
img = img.transpose(2, 0, 1)
文章知识点与官方知识档案匹配，可进一步学习相关知识
Python入门技能树人工智能深度学习416686 人正在系统学习中
pytorch：读取的图像变换通道
注：本文转载自blog.csdn.net的xddwz的文章"https://blog.csdn.net/xddwz/article/details/106549334"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
评论记录：
回复评论：