YOLOv5-Multibackbone-Compression icon indicating copy to clipboard operation
YOLOv5-Multibackbone-Compression copied to clipboard

Published 20 hours ago • verified publisher icon Gumpest

Transformer模块问题

Open GF1447571253 opened this issue 3 years ago • 6 comments

您好，我想问一下在common.py下的C3TR、TransformerBlock和TransformerLayer类是简化了吗？我在您的debug.py看到了您注释一个Transformer模块，两者之间好像不一样

GF1447571253 avatar Nov 15 '21 09:11 GF1447571253

是的，目的是减小过多的内存占用；例如减小通道数；当然C3TR也是YOLOv5官方提供的模块之一。

Gumpest avatar Nov 15 '21 09:11 Gumpest

我看您的代码只在最后一个预测头上用了C3TR，您尝试过四个预测头全部使用C3TR吗？

GF1447571253 avatar Nov 15 '21 12:11 GF1447571253

尝试过，效果不是很好；体现在显存占用过大以及精度较低（与YOLOv5x精度齐平）。

Gumpest avatar Nov 16 '21 08:11 Gumpest

尝试过，效果不是很好；体现在显存占用过大以及精度较低（与YOLOv5x精度齐平）。

您好，我想请问用一块3060显卡12G显存可以跑您这个模型吗，用的是yolov5m，yolov5TPH_std，batch-size调成1，还提示显存不足。还是需要我重新配置环境啊，目前我是在windows系统下跑的

Ziran-Li avatar Nov 17 '21 09:11 Ziran-Li

尝试过，效果不是很好；体现在显存占用过大以及精度较低（与YOLOv5x精度齐平）。

您好，我想请问用一块3060显卡12G显存可以跑您这个模型吗，用的是yolov5m，yolov5TPH_std，batch-size调成1，还提示显存不足。还是需要我重新配置环境啊，目前我是在windows系统下跑的

您好 可以尝试yolov5xTPH.yaml

Gumpest avatar Nov 24 '21 05:11 Gumpest

尝试过，效果不是很好；体现在显存占用过大以及精度较低（与YOLOv5x精度齐平）。

那请问您试过将backbone中多个C3替换成C3STR后的效果嘛？您觉得有这个必要去做这个实验嘛？

ruoyuchen0903 avatar Mar 26 '22 09:03 ruoyuchen0903