RuntimeError: output with shape [1, 28, 28] doesn‘t match the broadcast shape [3, 28, 28]
**
RuntimeError: output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28]
**
RuntimeError: output with shape [1, 208, 256] doesn’t match the broadcast shape [3, 208, 256]
pytorch下跑裂缝类代码运行时遇见shape不匹配问题。
原因:图片格式是灰度图只有一个channel,需要变成RGB图才可以
翻译
2021-01-13 21:36:10 ·
928 阅读 ·
1 评论