Update INT8 ONNX Models (Calibration done with cappe-5 trainset)

yolo_nas_pose_l_int8.onnx

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

yolo_nas_pose_m_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:62ff9da802e83719d25b2e0040d2bac6c6e8bcce0904dc1e1be2a2eb397add97
-size 156781680
+oid sha256:b3758c7d74afeb66728bd84080904952ad49bd12d9aad55afb5768a63b382cd3
+size 156781884

yolo_nas_pose_n_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1240ae85ecb3abae0ff8e32aa24da73da1d91bd386113fe8ce70b9ffe9f9b9c6
-size 28218615
+oid sha256:3ab78595d3d0452820cf2b21da4a9a7bc14aa2ff9dd2c9611d17192f385686a7
+size 28218717

yolo_nas_pose_s_int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:371f621790e231135ba258da2f5607d0df9c4505109cd07f4bb1a392dee03da2
-size 62288937
+oid sha256:33fce48f8418a959a384e877c2771bd50409be9edd8f18fcef0e46b0a5aa38a8
+size 62289039

yolo_nas_pose_to_onnx.py

@@ -127,7 +127,7 @@ image = cv2.resize(image, (input_image_shape[1], input_image_shape[0]))
 image_bchw = np.transpose(np.expand_dims(image, 0), (0, 3, 1, 2))
 
 # Prepare Calibration Dataset for INT8 Quantization
-dataset = load_dataset("cppe-5", split="test")
+dataset = load_dataset("cppe-5", split="train")
 hf_dataset_wrapper = HFDatasetWrapper(dataset, transform=preprocess)
 calibration_loader = DataLoader(hf_dataset_wrapper, batch_size=8)