2019年12月6日
pytorch与numpy中的通道交换问题
摘要： pytorch网络输入图像的格式为（C, H, W)，而numpy中的图像的shape为（H,W,C） 所以一般需要变换通道，将numpy中的shape变换为torch中的shape。 方法如下： # A是numpy数据类型A = A.transpose(0,1,2) # 没有改变：(h,w,c) 阅读全文
posted @ 2019-12-06 11:24 青牛梦旅行 阅读(4429) 评论(0) 推荐(0)