Diff of /src/subject_info.py [000000] .. [c1b1c5]

Switch to side-by-side view

--- a
+++ b/src/subject_info.py
@@ -0,0 +1,54 @@
+# create a folder for the registered subject
+
+import os
+import shutil
+import pickle
+
+SUBJECT_NAME = 'benny_cai'
+
+SCAN_POSE = 'side'
+# SCAN_POSE = 'front'
+
+if __name__ == '__main__':
+
+    folder_path = 'data/' + SUBJECT_NAME
+    if not os.path.isdir(folder_path):
+        os.mkdir(folder_path)
+    os.chdir(folder_path)
+
+    if not os.path.isdir(SCAN_POSE):
+        os.mkdir(SCAN_POSE)
+    os.chdir(SCAN_POSE)
+
+    for subfolder in ['color_images',
+                      'depth_images',
+                      'intrinsics',
+                      'OpenPose',
+                      'ViTPose_base',
+                      'ViTPose_large']:
+        if not os.path.isdir(subfolder):
+            os.mkdir(subfolder)
+
+    # copy odometry.log from 'data' folder
+    src = '../../odometry.log'
+    dst = 'odometry.log'
+    shutil.copyfile(src, dst)
+
+    file1 = open("../../subject_name.txt", "a")
+    file1.write(SUBJECT_NAME + '_' + SCAN_POSE + "\n")
+    file1.close()
+
+    for subfolder in ['OpenPose',
+                      'ViTPose_base',
+                      'ViTPose_large']:
+
+        for subsubfolder in ['keypoints', 'output_images']:
+            if not os.path.isdir(subfolder + '/' + subsubfolder):
+                os.mkdir(subfolder + '/' + subsubfolder)
+
+
+
+
+
+
+