yolov3训练voc数据集报错RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.

1.代码来自:https://github.com/eriklindernoren/PyTorch-YOLOv3; 2.主要原因是yolov3输出三个分支,voc类别只有20类,因为只改动了一个分支,所以出现最后三个分支的输出张量因为维度不对,而导致无法拼接; 3.解决方法:在yolov3.cfg中更改配置,共有三处: 第一处: 第二处: 第三处: 以上三处改动地方都一样,convolutiona
相关文章
相关标签/搜索