Diff of /compute_targets_all.py [000000] .. [c1b1c5]

Switch to side-by-side view

--- a
+++ b/compute_targets_all.py
@@ -0,0 +1,50 @@
+# Compute 3 targets for all subjects using all 3 HPE models
+
+import os
+import subprocess
+
+# Optimized ratios
+# ratios = {
+#     'ViTPose_large': {'target1': [0.29194393, 0.12326672], 'target2': [0.27588455, 0.57679719], 'target4': [0.3438, 0.1083]},
+#     'ViTPose_base': {'target1': [0.29466347, 0.12381275], 'target2': [0.28210566, 0.583985], 'target4': [0.3536, 0.0894]},
+#     'OpenPose': {'target1': [0.29612489, 0.11958833], 'target2': [0.29058278, 0.57921349], 'target4': [0.3420, 0.0516]}
+# }
+
+# initial ratios
+ratios = {
+    'ViTPose_large': {'target1': [0.3, 0.1], 'target2': [0.3, 0.55], 'target4': [0.35, 0.1]},
+    'ViTPose_base': {'target1': [0.3, 0.1], 'target2': [0.3, 0.55], 'target4': [0.35, 0.1]},
+    'OpenPose': {'target1': [0.3, 0.1], 'target2': [0.3, 0.55], 'target4': [0.35, 0.1]}
+}
+
+
+if __name__ == '__main__':
+
+    data_path = 'src/data'
+
+    # loop through the data folder
+    for SUBJECT_NAME in os.listdir(data_path):
+        print("subject: ", SUBJECT_NAME)
+        subject_folder_path = os.path.join(data_path, SUBJECT_NAME)
+        if os.path.isfile(subject_folder_path):
+            continue
+
+        # loop through two poses (front and side)
+        for SCAN_POSE in os.listdir(subject_folder_path):
+            print("scan pose: ", SCAN_POSE)
+
+            # loop through three HPE models to compute targets
+            for POSE_MODEL in ['OpenPose', 'ViTPose_base', 'ViTPose_large']:
+                print("pose model: ", POSE_MODEL)
+                subprocess.run([
+                    "python", "src/compute_target.py",
+                    "--pose_model={}".format(POSE_MODEL),
+                    "--subject_name={}".format(SUBJECT_NAME),
+                    "--scan_pose={}".format(SCAN_POSE),
+                    "--target1_r1", str(ratios[POSE_MODEL]['target1'][0]),
+                    "--target1_r2", str(ratios[POSE_MODEL]['target1'][1]),
+                    "--target2_r1", str(ratios[POSE_MODEL]['target2'][0]),
+                    "--target2_r2", str(ratios[POSE_MODEL]['target2'][1]),
+                    "--target4_r1", str(ratios[POSE_MODEL]['target4'][0]),
+                    "--target4_r2", str(ratios[POSE_MODEL]['target4'][1])
+                ])
\ No newline at end of file