bug:
TypeError: __init__() missing 1 required positional argument: 'c2'
我改进yolov7参考的是:
改进YOLOv7系列:23.YOLOv7添加CBAM注意力机制_yolov7 cbam-CSDN博客
YOLOv7改进:CBAM注意力机制_cbam注意力机制改进-CSDN博客
这两篇的yaml中还用到了CNeB模块和C3C2模块,于是我又参考了:
YOLOv7改进:ConvNeXt(backbone改为CNeB)_yolov7的改进bakebone-CSDN博客
解决办法:
在yolo.py的那个if中添加了C3C2和CNeB模块,如下图。
总之就是把yaml中用到的自己定义的模块(除了CBAM)加进去。
不懂具体原因,但确实解决了问题。