数据增强导致图片维度不匹配，求助？

首页课程实战体系课手记专栏慕课教程

数据增强导致图片维度不匹配，求助？

来源：6-15 PyTorch搭建cifar10推理测试脚本搭建

baby猫

2021-07-02

老师，黑白图片通道=1，使用这几种数据增强方式（Compose里注释掉的部分）后，训练时就说维度不匹配RuntimeError: output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28]，改网络输入通道数=3也没用。这究竟是怎么回事？
图片描述

写回答

1回答

会写代码的好厨师

会写代码的好厨师

2021-07-09

已采纳

确认下是否有预训练模型的加载
网络的第一层是不是输入层改成了单通道
num_output_channels 这个参数设置为3 灰度图也输出三通道

0

1

baby猫

非常感谢！

2021-07-10

共1条回复

PyTorch入门到进阶实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发，快速掌握PyTorch框架

1190 学习 · 298 问题

相似问题

train_transform 中使用了 Greyscale 会改变图片的输入维度么？

回答 1

加入的隐藏层中100会报错，由于496*13*13*1才能匹配，这里应该是13吧

回答 1

bug报错不一样

回答 2

高维的索引和数据筛选感觉不是很好理解

回答 1

二维张量第0维排序是根据什么排的?

回答 1

打开慕课网App查看更多内容