Rename onnx_eval.py to eval_onnx.py

onnx_eval.py → eval_onnx.py

@@ -87,7 +87,7 @@ def run(data,
         exist_ok=False,  # existing project/name ok, do not increment
         half=True,  # use FP16 half-precision inference
         plots=False,
-        onnx_weights="./yolov5s_qat.onnx",
+        onnx_model="./yolov5s.onnx",
         ipu=False,
         provider_config='',
         ):
@@ -99,14 +99,14 @@ def run(data,
     (save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True)  # make dir
 
     # Load model
-    if isinstance(onnx_weights, list):
-        onnx_weights = onnx_weights[0]
+    if isinstance(onnx_model, list):
+        onnx_model = onnx_model[0]
     if ipu:
         providers = ["VitisAIExecutionProvider"]
         provider_options = [{"config_file": provider_config}]
-        onnx_model = onnxruntime.InferenceSession(onnx_weights, providers=providers, provider_options=provider_options)
+        onnx_model = onnxruntime.InferenceSession(onnx_model, providers=providers, provider_options=provider_options)
     else:
-        onnx_model = onnxruntime.InferenceSession(onnx_weights)
+        onnx_model = onnxruntime.InferenceSession(onnx_model)
 
     # Data
     data = check_dataset(data)  # check
@@ -255,7 +255,7 @@ def parse_opt():
     parser.add_argument('--name', default='exp', help='save to project/name')
     parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
     parser.add_argument('--half', action='store_true', help='use FP16 half-precision inference')
-    parser.add_argument('-m', '--onnx_weights', default='./yolov5s_qat.onnx', nargs='+', type=str, help='path to your onnx_weights')
+    parser.add_argument('--onnx_model', default='./yolov5s.onnx', nargs='+', type=str, help='path to your onnx_weights')
     parser.add_argument('--ipu', action='store_true', help='flag for ryzen ai')
     parser.add_argument('--provider_config', default='', type=str, help='provider config for ryzen ai')
     opt = parser.parse_args()