Update 1.py

1.py

@@ -13,15 +13,15 @@ def set_prompt_and_get_coordinates(output_video, texts=['men', 'the table']):
         print(texts)
     with open('/kaggle/texts.pkl', 'wb') as file:
         pickle.dump(texts, file)
-    with open('/kaggle/output_video2.pkl', 'wb') as file:
+    with open('/kaggle/output_video1.pkl', 'wb') as file:
         pickle.dump(output_video, file)
-    command = ['python', '/kaggle/florence-sam-kaggle/kaggle_florence_gpu_2.py']
+    command = ['python', '/kaggle/florence-sam-kaggle/kaggle_florence_gpu_1.py']
     all_ok_bboxes = subprocess.run(command, capture_output=True, text=True)
     return all_ok_bboxes
 
 # 运行 sam2 处理
 def run_sam2(output_video):
-    script_path = '/kaggle/florence-sam-kaggle/kaggle_sam2_gpu_2.py'
+    script_path = '/kaggle/florence-sam-kaggle/kaggle_sam2_gpu_1.py'
     command = ['python3', script_path]
     sam2_output = subprocess.run(command, capture_output=True, text=True)
     print(sam2_output)
@@ -53,7 +53,7 @@ def create_video_with_audio(image_folder, input_video_path):
     os.makedirs(os.path.dirname(output_video_path), exist_ok=True)
 
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    video_writer = cv2.VideoWriter('/kaggle/image_sequence_video2.mp4', fourcc, fps, (width, height))
+    video_writer = cv2.VideoWriter('/kaggle/image_sequence_video1.mp4', fourcc, fps, (width, height))
 
     for image_file in image_files:
         image_path = os.path.join(image_folder, image_file)
@@ -62,7 +62,7 @@ def create_video_with_audio(image_folder, input_video_path):
 
     video_writer.release()
 
-    temp_video_path = '/kaggle/image_sequence_video2.mp4'
+    temp_video_path = '/kaggle/image_sequence_video1.mp4'
     command = [
         'ffmpeg',
         '-y',  # 覆盖输出文件
@@ -98,7 +98,7 @@ def process_all_videos(source_dir, target_dir, image_folder):
     video_files = [f for f in os.listdir(source_dir) if f.endswith(('.mp4', '.avi', '.mov'))]
     video_files.sort(key=natural_sort_key)
     # 反转列表
-    video_files.reverse()
+    #video_files.reverse()
     print(video_files)
     for video_file in video_files:
         video_path = os.path.join(source_dir, video_file)
@@ -124,6 +124,6 @@ def process_all_videos(source_dir, target_dir, image_folder):
 # 示例调用
 source_dir = '/kaggle/o_videos'
 target_dir = '/kaggle'
-image_folder = '/kaggle/output2'
+image_folder = '/kaggle/output1'
 
 process_all_videos(source_dir, target_dir, image_folder)