hukai97 / yolov5-5.x-annotations Goto Github PK

您好，感谢您对U版代码的详细注释。关于CBAM的使用，我有一些问题，如果您方便解答的话，我将十分感激
1、我希望在yolov3中插入CBAM，就正常的在yolo head前插入，但是通过yaml我没有办法非常清晰的观察yolo的结构，且照你的Yolov5s中插入，插入如下，不知是否正确

2、在yolo.py部分增加CBAM代码处，您的代码给出了SE的部分，但是抱歉我的水平有限，无法举一反三写出CBAM对应的操作，希望如果方便的话，也请赐教

3、您是否有跑过加入CBAM后的yolo，对mAP大约有多大的提升

!python /content/drive/MyDrive/yolov5-5.x-annotations-main/train.py --img 640 --batch 16 --epochs 200 --data /content/drive/MyDrive/YOLO/yuanshiyolov5peizhi/yuan.yaml --cfg /content/drive/MyDrive/yolov5-5.x-annotations-main/models/yolov5s.yaml --weights /content/drive/MyDrive/YOLO/yolov5/yolov5s.pt

Traceback (most recent call last):
File "/content/drive/MyDrive/yolov5-5.x-annotations-main/train.py", line 36, in
import val # for end-of-epoch mAP
File "/content/drive/MyDrive/yolov5-5.x-annotations-main/val.py", line 23, in
from models.experimental import attempt_load
File "/content/drive/MyDrive/yolov5-5.x-annotations-main/models/experimental.py", line 7, in
from models.common import Conv, DWConv
File "/content/drive/MyDrive/yolov5-5.x-annotations-main/models/common.py", line 17, in
from utils.datasets import letterbox
File "/content/drive/MyDrive/yolov5-5.x-annotations-main/utils/datasets.py", line 20, in
matplotlib.use('TkAgg')
File "/usr/local/lib/python3.7/dist-packages/matplotlib/cbook/deprecation.py", line 296, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.7/dist-packages/matplotlib/cbook/deprecation.py", line 358, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.7/dist-packages/matplotlib/init.py", line 1281, in use
plt.switch_backend(name)
File "/usr/local/lib/python3.7/dist-packages/matplotlib/pyplot.py", line 237, in switch_backend
newbackend, required_framework, current_framework))
ImportError: Cannot load backend 'TkAgg' which requires the 'tk' interactive framework, as 'headless' is currently running

博主你好，请问我想要在yaml文件中改网络格式，想要输入的是两个模态的数据（红外和可见光模态）。现在需要有两个输入的数据集要被读到，博主可以给一些建议吗，有偿，十分感谢了。

在utils.general.py文件中, 第420行的download函数, 项目中注释是没有用到, 其实是用到了
在data目录下的VOC.yaml文件的download配置中(63行), 调用了这个函数, 作用是下载VOC数据集并解压.
希望补充一下

谢谢up主的分享，不知道up主有没有把CAattention加进去试过，我把CoorAttention加进去后，运行yolo.py里的
if opt.profile:
img = torch.rand(8 if torch.cuda.is_available() else 1, 3, 640, 640).to(device)
print('image size is', img.size())
y = model(img, profile=True)
会报错：
kernel = torch.DoubleTensor([*(x[0].shape[2:])]) // torch.DoubleTensor(list((m.output_size,))).squeeze()
TypeError: must be real number, not NoneType，是从x_h = self.pool_h(x) # h avg pool过来的。
如果up主试过，不知道是否遇到这个问题？遇到这个问题是如何解决的？
谢谢！！！

比如说backbone内spp结构输出【1,512,20,20】，1具体指代哪一个量的输出？512是否为channel输出？