Update align.py

align.py

@@ -126,7 +126,7 @@ def create_aligned_videos(seq1_frames, seq2_frames, alignment_path, output_combi
     fourcc = cv2.VideoWriter_fourcc(*'H264')
 
     # 创建并排视频
-    combined_out = cv2.VideoWriter(output_combined_path, fourcc, fps, (unified_width * 2, unified_height))
+    # combined_out = cv2.VideoWriter(output_combined_path, fourcc, fps, (unified_width * 2, unified_height))
     # 创建单独的视频
     seq1_out = cv2.VideoWriter(output_seq1_path, fourcc, fps, (unified_width, unified_height))
     seq2_out = cv2.VideoWriter(output_seq2_path, fourcc, fps, (unified_width, unified_height))
@@ -148,17 +148,17 @@ def create_aligned_videos(seq1_frames, seq2_frames, alignment_path, output_combi
         combined_frame = np.hstack((frame1_resized, frame2_resized))
 
         # 写入到输出视频
-        combined_out.write(combined_frame)
+        # combined_out.write(combined_frame)
         seq1_out.write(frame1_resized)
         seq2_out.write(frame2_resized)
 
     # 释放资源
-    combined_out.release()
+    # combined_out.release()
     seq1_out.release()
     seq2_out.release()
 
 
-def align_filter(standard, user):
+def align_filter(standard, user, tmpdir):
     # 示例数据
     sequence1 = json.load(open(standard + ".json", 'r'))
     sequence2 = json.load(open(user + ".json", 'r'))
@@ -181,7 +181,7 @@ def align_filter(standard, user):
     seq2_frames = read_video_frames(user + '.mp4')  # Sequence 2: 34 frames
 
     # 输出视频路径
-    output_combined_path = 'tmp/aligned_combined_output.mp4'  # 合并视频
+    output_combined_path = tmpdir + '/aligned_combined_output.mp4'  # 合并视频
     output_seq1_path = standard + '.mp4'          # 对齐后sequence1视频
     output_seq2_path = user + '.mp4'          # 对齐后sequence2视频
 
@@ -195,6 +195,6 @@ def align_filter(standard, user):
     # 创建对齐后的视频
     create_aligned_videos(seq1_frames, seq2_frames, alignment_path, output_combined_path, output_seq1_path, output_seq2_path)
 
-    print(f"Combined video created at {output_combined_path}")
+    # print(f"Combined video created at {output_combined_path}")
     print(f"Aligned Sequence 1 video created at {output_seq1_path}")
     print(f"Aligned Sequence 2 video created at {output_seq2_path}")