解决:yolov7 RuntimeError: Given groups=1, weight of size [1, 64, 1, 1], expected input[1, 256, 64, 64]

1. 问题:在添加注意力机制时,出现问题。

2. 解决:在yolo.py文件中进行修改

 

 将填入的注意力机制SEAttention,改为相对应yolov7.yaml修改的名称。

猜你喜欢

转载自blog.csdn.net/weixin_44813538/article/details/131853785