Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

<neck>change concat to weightedadd

Browse Source
This commit is contained in:
Zhang Wei committed 2025-12-25 09:25:37 +08:00
1 parent 434a7ade3f
commit 492ad7a7fe
2 files changed
+27 -23
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

models/yolov5s.yaml
+24 -19
View File
@@ -49,26 +49,31 @@ backbone:
# ]
# YOLOv5 v6.0 head (PAN + WeightedAdd)
head: [
# ---------- Top-down FPN ----------
[-1, 1, Conv, [512, 1, 1]], # 10: reduce P5
[-1, 1, nn.Upsample, [None, 2, "nearest"]], # 11
[[-1, 6], 1, WeightedAdd, [2]], # 12: P5 + backbone P4
[-1, 3, C3, [512, False]], # 13
head:
[
# ---------- Top-down path ----------
[-1, 1, Conv, [512, 1, 1]], # 10 reduce P5
[-1, 1, nn.Upsample, [None, 2, "nearest"]], # 11
[6, 1, Conv, [512, 1, 1]], # 12 align backbone P4
[[-1, -2], 1, WeightedAdd, [2]], # 13 P5 + P4
[-1, 3, C3, [512, False]], # 14
[-1, 1, Conv, [256, 1, 1]], # 14: reduce
[-1, 1, nn.Upsample, [None, 2, "nearest"]], # 15
[[-1, 4], 1, WeightedAdd, [2]], # 16: P4 + backbone P3
[-1, 3, C3, [256, False]], # 17 (P3/8)
[-1, 1, Conv, [256, 1, 1]], # 15 reduce
[-1, 1, nn.Upsample, [None, 2, "nearest"]], # 16
[4, 1, Conv, [256, 1, 1]], # 17 align backbone P3
[[-1, -2], 1, WeightedAdd, [2]], # 18 P4 + P3
[-1, 3, C3, [256, False]], # 19 (P3/8)
# ---------- Bottom-up PAN ----------
[-1, 1, Conv, [256, 3, 2]], # 18: downsample
[[-1, 13], 1, WeightedAdd, [2]], # 19: P3 -> P4
[-1, 3, C3, [512, False]], # 20 (P4/16)
# ---------- Bottom-up path ----------
[-1, 1, Conv, [256, 3, 2]], # 20 downsample P3 -> P4
[14, 1, Conv, [256, 1, 1]], # 21 align top-down P4
[[-1, -2], 1, WeightedAdd, [2]], # 22
[-1, 3, C3, [512, False]], # 23 (P4/16)
[-1, 1, Conv, [512, 3, 2]], # 21: downsample
[[-1, 10], 1, WeightedAdd, [2]], # 22: P4 -> P5
[-1, 3, C3, [1024, False]], # 23 (P5/32)
[-1, 1, Conv, [512, 3, 2]], # 24 downsample P4 -> P5
[10, 1, Conv, [512, 1, 1]], # 25 align P5
[[-1, -2], 1, WeightedAdd, [2]], # 26
[-1, 3, C3, [1024, False]], # 27 (P5/32)
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
]
[[19, 23, 27], 1, Detect, [nc, anchors]],
]
train_my_gra.sh
+3 -4
View File
@@ -2,9 +2,8 @@ export CUDA_VISIBLE_DEVICES=4,5,6,7
python /data6/zhangw/yolov5/train.py \
--img 960 \
--batch 16 \
--epochs 300 \
--epochs 500 \
--data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
--weights /data6/zhangw/yolov5/yolov5s6.pt \
--project /data6/zhangw/ZW_gra_project/yolov5_result \
--project /data6/zhangw/ZW_gra_project/yolov5_result_structure \
--cfg /data6/zhangw/yolov5/models/yolov5s.yaml \
--name G2_2_cub100_yolov5s6_changeNeck_e300
--name G2_2_cub100_yolov5s6_changeNeck_e500