fix perfect test

Author: ccssu

oneflow_train_test/config.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 One_YOLOv5_ROOT=/home/fengwen/one-yolov5
-Weights=/home/fengwen/one-yolov5/runs/yolov5n
+Weights=/home/fengwen/one-yolov5/runs/yolov5s
 Device=0 # cuda:0
-Model=yolov5n
+Model=yolov5s
 Test_Ci_Project='test_ci_project'

oneflow_train_test/train_all_test.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
+set -e 
 
 bash train_default_test.sh config.sh 
 

oneflow_train_test/train_cls_test.sh

@@ -1,5 +1,4 @@
 #!/bin/bash
-
 source $1
 
 set -e # Error interrupt encountered
@@ -13,10 +12,11 @@ rm $ProjectPath -rf
 
 echo "Test classify"
 b=$ProjectPath/exp/weights/best  # best.of checkpoint
-python train.py --imgsz 64 --batch 32 --weights ${Weights}-cls.of --epochs 1  --project $ProjectPath               # train
 
-python val.py  --batch 32 --weights ${b}.of --device ${Device}                                                      # val
+python classify/train.py  --imgsz 64 --batch 32 --model ${Weights}-cls.of --epochs 1  --project $ProjectPath      # train
+
+python classify/val.py  --batch 32 --weights ${b}.of --device ${Device} --data  /home/fengwen/datasets/imagenette160                                             # val
 
-python detect.py --weights ${b}.of --device ${Device}                                                               # detect
+python classify/predict.py --weights ${b}.of --device ${Device}   --source data/images/                                      # detect
 
 python export.py --weights $b.of --img 64 --include onnx --device ${Device}                                         # export