search

Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
989 stars 201 forks source link

Transformer模块问题 #10

Open GF1447571253 opened 2 years ago

GF1447571253 commented 2 years ago

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

Gumpest commented 2 years ago

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

GF1447571253 commented 2 years ago

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

Gumpest commented 2 years ago

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

Ziran-Li commented 2 years ago

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

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

Gumpest commented 2 years ago

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

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

您好 可以尝试yolov5xTPH.yaml

ruoyuchen0903 commented 2 years ago

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

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