目标检测评估数据集transform问题
收藏
读取验证集数据的时候,在进行数据transform的时候报了如下错误
训练的时候都能正常增强,进行transform,但是预测的时候就报了这个错误,是什么原因
目前定位到是在进行batch合并的时候好像维度出错了,但具体不知道怎么解决。
0
收藏
请登录后评论
你把你的每一个图片的size看一下,应该是有不一样的。channel和size都要一致。