diff --git a/export_my_gra.sh b/export_my_gra.sh index 953f51e..7531eb8 100644 --- a/export_my_gra.sh +++ b/export_my_gra.sh @@ -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/ZW_gra_project/yolov5_result_pretrained/G12_cls_head_enhance_ECA_e300/weights/best.pt \ + --img 640 \ --batch 1 \ --device cpu \ --include onnx \