RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 544 and 193

解决方案汇总:(持续更新) 参考了多位博主的文章,仅作学习用,如有侵权请联系我删除。 一、原因 二、解决方法 1.修改dataloader源码: 链接:here 2.修改padding使图片size相同 在maskrcnn_benchmark里,image_list类,后面找到源码再来更新。
相关文章
相关标签/搜索