cls_head_enhance_and_ECA

This commit is contained in:
Zhang Wei committed 2026-03-20 13:38:48 +08:00
1 parent 5dbde58605
commit 0bf2989264
3 files changed
+191 -5

No files matched your search

+2 -2
View File
@@ -6,5 +6,5 @@ python /data6/zhangw/yolov5/train.py \
--data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
--weights /data6/zhangw/yolov5/yolov5s6.pt \
--project /data6/zhangw/ZW_gra_project/yolov5_result_pretrained \
--name G11_cls_head_enhance_Lightweight_e300 \
--cfg /data6/zhangw/yolov5/models/yolov5s6_cls_enhance.yaml
--name G12_cls_head_enhance_ECA_e300 \
--cfg /data6/zhangw/yolov5/models/yolov5s6_cls_enhance_ECA.yaml