Open GF1447571253 opened 2 years ago
是的,目的是减小过多的内存占用;例如减小通道数;当然C3TR也是YOLOv5官方提供的模块之一。
我看您的代码只在最后一个预测头上用了C3TR,您尝试过四个预测头全部使用C3TR吗?
尝试过,效果不是很好;体现在显存占用过大以及精度较低(与YOLOv5x精度齐平)。
尝试过,效果不是很好;体现在显存占用过大以及精度较低(与YOLOv5x精度齐平)。
您好,我想请问用一块3060显卡12G显存可以跑您这个模型吗,用的是yolov5m,yolov5TPH_std,batch-size调成1,还提示显存不足。还是需要我重新配置环境啊,目前我是在windows系统下跑的
尝试过,效果不是很好;体现在显存占用过大以及精度较低(与YOLOv5x精度齐平)。
您好,我想请问用一块3060显卡12G显存可以跑您这个模型吗,用的是yolov5m,yolov5TPH_std,batch-size调成1,还提示显存不足。还是需要我重新配置环境啊,目前我是在windows系统下跑的
您好 可以尝试yolov5xTPH.yaml
尝试过,效果不是很好;体现在显存占用过大以及精度较低(与YOLOv5x精度齐平)。
那请问您试过将backbone中多个C3替换成C3STR后的效果嘛?您觉得有这个必要去做这个实验嘛?
您好,我想问一下在common.py下的C3TR、TransformerBlock和TransformerLayer类是简化了吗?我在您的debug.py看到了您注释一个Transformer模块,两者之间好像不一样