[G1][Backbone] add CBAM and export scripts

This commit is contained in:
Zhang Wei committed 2025-12-18 15:40:26 +08:00
1 parent 85acef3a28
commit bc7d124645
17 files changed
+1044 -95

No files matched your search

+10
View File
@@ -0,0 +1,10 @@
export CUDA_VISIBLE_DEVICES=0,1,2,3
python /data6/zhangw/yolov5/train.py \
--img 960 \
--batch 16 \
--epochs 60 \
--data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
--weights /data6/zhangw/yolov5/yolov5s6.pt \
--project /data6/zhangw/ZW_gra_project/yolov5_result \
--name G1_cub100_yolov5s6_changeBackbone_e60 \
--cfg /data6/zhangw/yolov5/models/yolov5s.yaml