Regenerate ONNX models

yolo_nas_pose_l_fp16.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d4da0316c679b93931da1a30d1224d280991d2350354cb8df7338b0affc154e3
-size 108987993
+oid sha256:de6509c04ab507c7808f217f39004b0ba7f0aaf9cc23efbfc8e05adc2694326b
+size 108988039

yolo_nas_pose_l_fp16.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_l_fp32.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cfd7c12de1275152ce64af33b734c74c085ed45403fb2722385718850a71f363
-size 217830699
+oid sha256:b70925839e17c877be91e7f681fe8eb425ccc89edc6e9cd821a4a246b32e6b6e
+size 217830745

yolo_nas_pose_l_fp32.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_l_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7466a74638b064b0abef6722ca4744d1c7d07f47177dbf9d8bb06c492e4988d4
-size 218735452
+oid sha256:c8e5a37af669630e0d718bb0be9bc0d1dada1dc977fcd5235f63b678480dbf6b
+size 218735498

yolo_nas_pose_l_int8.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_m_fp16.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7674809abd50d8acbab6500f1b7ad6cb0103539e6102066d7695160b9b0f8413
-size 78063545
+oid sha256:890495a6b1a9ed8bd041c47143c03ba831a78b143852d6e150617bf0d31a2720
+size 78063591

yolo_nas_pose_m_fp16.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_m_fp32.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d535c1b16741c82edb81d0eaf11604ed2c4cd4e094a3cedb0ac4d175885b0502
-size 155996929
+oid sha256:e7b86be1fdfd418b58a9db77c601fca8241310b7988098be63f3b7e2f3b0d2e6
+size 155996975

yolo_nas_pose_m_fp32.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_m_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b3758c7d74afeb66728bd84080904952ad49bd12d9aad55afb5768a63b382cd3
-size 156781884
+oid sha256:731d5c1229bf3694a14bc542b2a906ab138d9a6a8bef9227738940afc650fee5
+size 156781930

yolo_nas_pose_m_int8.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_n_fp16.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7780a9bdc0a3302f98f72cdec9c631f2d6d9c347c2dceefc314343235ac4a280
-size 13909326
+oid sha256:376f5d1216a9ef226d99eb534e838d60f572afb5620cc0bc3249282063def264
+size 13909372

yolo_nas_pose_n_fp16.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_n_fp32.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7868a9dbe652ba34d13ad0502bfeb9abdf00a8e11315d98f03f7b58b6bcfb181
-size 27697216
+oid sha256:0bdb77b9648c168466f0a95925cd33db5c6d52602357576567c05bc19f080ac5
+size 27697262

yolo_nas_pose_n_fp32.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_n_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ab78595d3d0452820cf2b21da4a9a7bc14aa2ff9dd2c9611d17192f385686a7
-size 28218717
+oid sha256:09467de3b1e52a8de31d31b46eb2c2588dc9e02acb35a0d578915d2a0d78890f
+size 28218763

yolo_nas_pose_n_int8.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_s_fp16.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fe40712b6dbff75170c0dbc9a5c9789532f4a9c29b592e361ff602dfa28b2f7e
-size 30900352
+oid sha256:d6b2b0397db57b09d289ac74167e5af7b1a08e8d3e3aaae0319ea02bd880fb93
+size 30900398

yolo_nas_pose_s_fp16.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_s_fp32.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6c940f729387bebd4f4ffc1f34df78f495ca7f0b4e8099e575c88ca5185e0f6c
-size 61676424
+oid sha256:9eb5dd7adee077ac9fe59917781361c33b6efce9ff0cb76654c577a017760bc9
+size 61676470

yolo_nas_pose_s_fp32.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_s_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:33fce48f8418a959a384e877c2771bd50409be9edd8f18fcef0e46b0a5aa38a8
-size 62289039
+oid sha256:9704c804bdf088fac2468a811924d345a3d9c687b4c8371745d8e6191bdb681f
+size 62289085

yolo_nas_pose_s_int8.onnx.usage.txt

@@ -15,7 +15,7 @@ Sequential(
 Exported model contains postprocessing (NMS) step with the following parameters:
     num_pre_nms_predictions=1000
     max_predictions_per_image=10
-    nms_threshold=0.5
+    nms_threshold=0.2
     confidence_threshold=0.15
     output_predictions_format=flat
 

yolo_nas_pose_to_onnx.py

@@ -160,7 +160,7 @@ for model_name in [Models.YOLO_NAS_POSE_L,  Models.YOLO_NAS_POSE_M,  Models.YOLO
 				engine=ExportTargetBackend.ONNXRUNTIME,
 				quantization_mode=q,
 				#selective_quantizer: Optional["SelectiveQuantizer"] = None,  # noqa
-				calibration_loader = calibration_loader,
+				calibration_loader = calibration_loader if q == ExportQuantizationMode.INT8 else None,
 				#calibration_method: str = "percentile",
 				#calibration_batches: int = 16,
 				#calibration_percentile: float = 99.99,