add decoupled head

2026-03-14 10:34:02 +08:00
1 parent 9b008eaa99
commit 14aeebc1e5
@@ -1,6 +1,6 @@
 python export.py \
-  --weights /data6/zhangw/ZW_gra_project/yolov5_result_pretrained/G0_cub100_yolov5s6_raw_e300/weights/best.pt \
-  --img 640 640 \
+  --weights /data6/zhangw/yolov5/yolov5s6.pt \
+  --img 640 \
  --batch 1 \
  --device cpu \
  --include onnx \