Limit opset version to its minimum, 14

yolo_nas_pose_to_onnx.py

@@ -30,7 +30,7 @@ output_predictions_format=DetectionOutputFormatMode.FLAT_FORMAT
 
 # NMS-related Setting
 confidence_threshold=.15
-nms_threshold=.5
+nms_threshold=.2
 num_pre_nms_predictions=1000
 max_predictions_per_image=10
 
@@ -172,7 +172,7 @@ for model_name in [Models.YOLO_NAS_POSE_L,  Models.YOLO_NAS_POSE_M,  Models.YOLO
 				#input_image_channels: Optional[int] = None,
 				#input_image_dtype: Optional[torch.dtype] = None,
 				max_predictions_per_image=max_predictions_per_image,
-				#onnx_export_kwargs: Optional[dict] = None,
+				onnx_export_kwargs={"opset_version":14},
 				onnx_simplify=True,
 				#device: Optional[Union[torch.device, str]] = None,
 				output_predictions_format=output_predictions_format,