m0_57814193的博客

原创 RuntimeError: Sizes of tensors must match except in dimension 1. Got 16 and 32 in dimension 2报错解决

比如，如果你有两个特征图，一个是 16×16，另一个是 32×32，那么你可以将 16×16 的特征图上采样到 32×32。在大多数情况下，当使用 YOLO 或类似的体系结构时，这种错误通常发生在特征图融合阶段，即当尝试将来自不同层级的特征图进行拼接时。通常，特征图拼接是沿着特征通道维度进行的（即维度1，对应PyTorch中的通道），但你需要确保所有操作前后都保持这一约定。在你的网络中，如果你在拼接不同的特征层（可能是不同的卷积层输出），你需要确保这些特征图在拼接前在高度和宽度上是匹配的。