Update evaluate.py

evaluate.py

@@ -67,18 +67,18 @@ def load_json(path):
     with open(path, 'r') as file:
         return json.load(file)
 
-def eval(test, standard):
-    test_p = "tmp/user.mp4"
-    standard_p = "tmp/standard.mp4"
-    os.system('python inferencer_demo.py ' + test_p) # produce user.json
+def eval(test, standard, tmpdir):
+    test_p = tmpdir + "/user.mp4"
+    standard_p = tmpdir + "/standard.mp4"
+    os.system('python inferencer_demo.py ' + test_p + ' --pred-out-dir', tmpdir) # produce user.json
 
     scores = []
 
-    align_filter('tmp/standard', 'tmp/user') # 帧对齐 produce aligned vedios
+    align_filter(tmpdir + '/standard', tmpdir + '/user', tmpdir) # 帧对齐 produce aligned vedios
 
     # return None
-    data_00 = load_json('tmp/standard.json') # aligned json
-    data_01 = load_json('tmp/user.json')
+    data_00 = load_json(tmpdir + '/standard.json') # aligned json
+    data_01 = load_json(tmpdir + '/user.json')
     cap_00 = cv2.VideoCapture(standard_p)
     cap_01 = cv2.VideoCapture(test_p)
     # Define keypoint connections for both videos (example indices, you'll need to customize)
@@ -91,7 +91,7 @@ def eval(test, standard):
     frame_width = int(cap_00.get(cv2.CAP_PROP_FRAME_WIDTH))
     frame_height = int(cap_00.get(cv2.CAP_PROP_FRAME_HEIGHT))
 
-    out = cv2.VideoWriter('tmp/output.mp4', cv2.VideoWriter_fourcc(*'H264'), 5, (frame_width*2, frame_height*2))
+    out = cv2.VideoWriter(tmpdir + '/output.mp4', cv2.VideoWriter_fourcc(*'H264'), 5, (frame_width*2, frame_height*2))
 
     cap_00.set(cv2.CAP_PROP_POS_FRAMES, 0) # 初始化视频从头开始读取
     cap_01.set(cv2.CAP_PROP_POS_FRAMES, 0)