Add comparison between joint coordinated from original and converted models

convert_to_mixed.py

@@ -55,6 +55,17 @@ def main(args):
     is_close = np.allclose(original_result[0], converted_result[0], rtol=args.rtol, atol=args.atol)
     print(f"Validation result: {'Success' if is_close else 'Failure'}")
 
+    print('Compare Joint Coordinates:')
+    original_keypoints, converted_keypoints = original_result[1][0, :, :, :2], converted_result[1][0, :, :, :2]
+    for n_det in range(original_keypoints.shape[0]):
+        print(f'\tDetection {n_det}:')
+        for n_joint in range(original_keypoints.shape[1]):
+            original_x, original_y = original_keypoints[n_det][n_joint]
+            converted_x, converted_y = converted_keypoints[n_det][n_joint]
+            print(f'\t\tJoint {n_joint}:')
+            print(f'\t\t\tX: {original_x:.2f}, {converted_x:.2f}')
+            print(f'\t\t\tY: {original_y:.2f}, {converted_y:.2f}')
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Convert an ONNX model to mixed precision format.")
     parser.add_argument("source_model_path", type=str, help="Path to the source ONNX model.")