path

ColorCorrection/color_correction.py

@@ -3,6 +3,7 @@
 from glob import glob
 import time
 import subprocess
+import json
 
 def run_extract_undistort(src_dir, FOV):
     proc = subprocess.Popen(['python', 'undistort.py', '--src_dir', src_dir, '--FOV', FOV])
@@ -17,23 +18,15 @@ def run_color_correction(root_dir, atals_dir, tar_idx):
     return proc
 
 
-# case_arr = [("./200307_hyomin/", "new_all2/")]
-case_arr = [("./210111_hyomin/", "test/")]
-# case_arr = [("./200304_police/", "test/"), ("./200627_shark_tilt/", "ttt/"), ("./200630_mario2/", "ttt1/"), ("./200629_sister3/", "ttt2/"), ("./200307_hyomin/", "swtest/")]
-# case_arr = [("./gunhee/", "new_all/"), ("./200304_police/", "new_all/"), ("./200627_shark_tilt/", "new_all/"), ("./200630_mario2/", "new_all/"), ("./200629_sister3/", "new_all/"), ("./200307_hyomin/", "new_all/")]
-# case_arr = [("./200629_sister3/", "finalfinal/")]
-# case_arr = [("./gunhee/", "global/"), ("./gunhee/", "multi/")]
-# case_arr = [("./gunhee/", "final2/")]
-# case_arr = [("./200627_shark_tilt/", "final/")]
-# case_arr = [("./200307_hyomin/", "finalfinal_pong2_3/")]
-# case_arr = [("./200307_hyomin/", "swtest/"), ("./200307_hyomin/", "swtest_nosw/"), ("./200307_hyomin/", "swtest_notemp/")]
-# case_arr = [("./200307_hyomin/", "swtest/")]
-# case_arr = [("./200630_mario2/", "ttt1/")]
-# case_arr = [("./200618_haeun_move2/", "ttt/")]
+with open('../TextureMappingNonRigid/conf.json', 'rt', encoding='UTF8') as f:
+    conf = json.load(f)
+
+case_arr = [("case_main_test")]
 
 for case in case_arr:
-    root_dir = case[0]
-    atals_dir = case[1]
+    data_case = conf[case]
+    root_dir = data_case['main']['data_root_path']
+    atals_dir = data_case['main']['atlas_path'] + '/'
     file_list = glob(root_dir + "atlas/" + atals_dir + "/video/Frame_*.png")
 
     count = len(file_list)

README.md

@@ -7,7 +7,7 @@ Spatiotemporal Texture Reconstruction
 ![Teaser](teaser.png)
 ### Requirements
 * Windows 10
-* CUDA >= 10.0
+* CUDA >= 10.2
 * python >= 3.6
 * MSVC++ >= 14.2
 * OpenCV >= 3.4

SHOT/shot.py

@@ -16,9 +16,7 @@ def run_color_correction(root_dir, tar_idx):
 with open('../TextureMappingNonRigid/conf.json', 'rt', encoding='UTF8') as f:
     conf = json.load(f)
 
-# case_array = ['case_haeun_18_move1', 'case_haeun_18_move2', 'case_hyeonseo_18_kick1', 'case_hyeonseo_18_kick2']
-# case_array = ['case_police']
-case_array = ['case_hyomin_01']
+case_array = ['case_main_test']
 
 
 for case in case_array:

Similarity/.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.pythonPath": "C:\\Users\\hyomin\\Anaconda3\\python.exe"
+}

Similarity/T2Utils.py

@@ -0,0 +1,15 @@
+import time
+
+
+def watchModule(timing_module, timing_all):
+    def watch(func):
+        def watch_f(*args, **kwargs):
+            if timing_module or timing_all:
+                start = time.time()
+            result = func(*args, **kwargs)
+            if timing_module or timing_all:
+                end = time.time()
+                print("%s %s %f" % (func.__name__, "elapse time:", end - start))
+            return result
+        return watch_f
+    return watch

Similarity/renderer.py

@@ -3,7 +3,6 @@
 import glm
 
 from GLmesh import *
-from GLpcd import *
 import ShaderLoader
 
 import numpy as np