xiaoyi-creator

奥奥，好的，还有一个问题想问您一下，就是原论文中的WBF代替NMS，我在您的代码中没有看到，想请问一下是您对这部分内容没有复现吗

您好，我在您的yolov5xYPH.yaml文件中看到你在预测头的c3模块之前加了SPP模块，想问一下这里加SPP模块是为了更好的提取特征吗？

我通过导入pip install timm的方式解决了这个问题，但是运行还是出现了新的问题 Traceback (most recent call last): File "train.py", line 542, in train(hyp, opt, device, tb_writer) File "train.py", line 88, in train model = Model(opt.cfg or ckpt['model'].yaml, ch=3, nc=nc,...

下载了您的代码来尝试SWinTr但是还是会报错 Sizes of tensors must match except in dimension 1. Expected size 24 but got size 19 for tensor number 1 in the list.，我尝试修改了一些东西错误消失，但是又出现了写的错误 File "E:\GoogleDownload\YOLOv5-Multibackbone-Compression-main\utils\datasets.py", line 388, in __init__...

使用您的代码训练 Epoch gpu_mem box obj cls labels img_size 0% 0/1701 [00:00

我使用了您的代码训练，租用了同等算力的GPU但是还是会报错 File "/user-data/yolov5-5.0/models/SwinTransformer.py", line 88, in forward x = (attn @ v).transpose(1, 2).reshape(B_, N, C) RuntimeError: expected scalar type Float but found Half 不知道为什么

按照您意见加上--nohalf之后又出现了新的错误 File "/user-data/YOLOv5-Multibackbone-Compression-main/models/SwinTransformer.py", line 242, in forward x_windows = window_partition(shifted_x, self.window_size) # nW*B, window_size, window_size, C File "/user-data/YOLOv5-Multibackbone-Compression-main/models/SwinTransformer.py", line 122, in window_partition x = x.view(B, H // window_size, window_size, W...

> 我直接运行昨天刚下载的您的仓库的代码就 可以跑通， 你说的跑的通是SwinTransformer可以跑通吗，我总是在第一轮训练计算map时出现错误

> 我下载了您的程序，但是我使用了自己的数据集，运行命令： python train.py --data my.yaml --weights yolov5s.pt --cfg models/accModels/yolov5xP2CBAM-Swin-BiFPN-SPP.yaml --hyp data/hyps/hyp.visdrone.yaml --epochs 60 --batch-size 2 --img 640 --nohalf 您觉得有可能是我数据集的问题吗 我百度查找相关错误，有说是因为nc的数值不对引起的，但是我检查了数值并没有问题