From c175666b38fd1d6f135262df85ef7aea0e92572f Mon Sep 17 00:00:00 2001
From: vicrrs <enricofermi2019@gmail.com>
Date: Fri, 31 Jan 2025 08:29:11 -0400
Subject: [PATCH] Fix ONNX file name to avoid .pt.onnx

---
 utils/export_yoloV8.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/export_yoloV8.py b/utils/export_yoloV8.py
index 9d78a183..5a14bc62 100644
--- a/utils/export_yoloV8.py
+++ b/utils/export_yoloV8.py
@@ -94,7 +94,7 @@ def main(args):
     img_size = args.size * 2 if len(args.size) == 1 else args.size
 
     onnx_input_im = torch.zeros(args.batch, 3, *img_size).to(device)
-    onnx_output_file = f'{args.weights}.onnx'
+    onnx_output_file = args.weights.replace(".pt", "") + ".onnx"
 
     dynamic_axes = {
         'input': {