Update all_in_one.py

all_in_one.py

@@ -7,19 +7,20 @@ import re
 import pickle
 from moviepy.editor import VideoFileClip, ImageSequenceClip
 import subprocess
-
+out_dir = "/workspace/tem"
+os.makedirs(out_dir, exist_ok=True)
 def set_prompt_and_get_coordinates(output_video, texts=['men', 'the table']):
     if isinstance(texts, str):
         texts = texts.split(',')  # Assuming prompts are separated by commas
         texts = [text.strip() for text in texts]
         print(texts)
     # 保存提示词到文件
-    with open('/workspace/florence-sam/texts.pkl', 'wb') as file:
+    with open('/workspace/tem/texts.pkl', 'wb') as file:
         pickle.dump(texts, file)
-    with open('/workspace/florence-sam/output_video.pkl', 'wb') as file:
+    with open('/workspace/tem/output_video.pkl', 'wb') as file:
         pickle.dump(output_video, file)
     # 构建命令
-    command = ['python', '/workspace/florence-sam/1.py']
+    command = ['python', '/workspace/florence-sam-tencent/1.py']
 
     # 执行命令并捕获输出
     all_ok_bboxes = subprocess.run(command, capture_output=True, text=True)
@@ -38,7 +39,7 @@ print(result.stdout)
 def run_sam2(output_video):
 
     # 定义脚本路径
-    script_path = '/workspace/florence-sam/2.py'
+    script_path = '/workspace/florence-sam-tencent/2.py'
 
     # 构建命令
     command = ['python3', script_path]
@@ -93,7 +94,7 @@ def create_video_with_audio(image_folder, input_video_path):
 
     # 生成与输入视频同名的输出文件
     #output_video_path = os.path.join('/tmp/gradio/', os.path.basename(input_video_path))
-    output_video_path = os.path.join('/workspace/florence-sam/sam2_videos/', os.path.basename(input_video_path))
+    output_video_path = os.path.join('/workspace/tem/sam2_videos/', os.path.basename(input_video_path))
     # 确保输出目录存在
     os.makedirs(os.path.dirname(output_video_path), exist_ok=True)
 
@@ -104,8 +105,8 @@ def create_video_with_audio(image_folder, input_video_path):
     return output_video_path
 
 # 示例调用
-image_folder = '/workspace/florence-sam/output'
-with open('/workspace/florence-sam/output_video.pkl', 'rb') as file:
+image_folder = '/workspace/tem/output'
+with open('/workspace/tem/output_video.pkl', 'rb') as file:
     output_video = pickle.load(file)
 print(output_video)
 input_video_path = output_video