File size: 1,529 Bytes
f8ba0eb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | import os
import sys
import argparse
VIDEO_FILE_DIR = "/home/xiuying/Code/xywang/test_videos"
# API服务器的URL
API_URL = "http://127.0.0.1:8010/video-inference/"
PROMPT = "视频里发生了什么？"
for file in os.listdir(VIDEO_FILE_DIR):
if file.endswith(".mp4"):
VIDEO_FILE_PATH = os.path.join(VIDEO_FILE_DIR, file)
command = (
f"curl -v -X POST '{API_URL}' "
f"-F 'prompt={PROMPT}' "
f"-F 'video_file=@{VIDEO_FILE_PATH};type=video/mp4' "
f"-F 'sampling_method=content_aware' "
f"-F 'sampling_rate=28' "
)
print("将要执行以下 cURL 命令:")
print("---------------------------------")
print(command)
print("---------------------------------")
print("\n正在执行...\n")
os.system(command)
for file in os.listdir(VIDEO_FILE_DIR):
if file.endswith(".mp4"):
VIDEO_FILE_PATH = os.path.join(VIDEO_FILE_DIR, file)
command = (
f"curl -v -X POST '{API_URL}' "
f"-F 'prompt={PROMPT}' "
f"-F 'video_file=@{VIDEO_FILE_PATH};type=video/mp4' "
f"-F 'sampling_method=uniform' "
f"-F 'sampling_rate=28' "
)
print("将要执行以下 cURL 命令:")
print("---------------------------------")
print(command)
print("---------------------------------")
print("\n正在执行...\n")
os.system(command)
print("\n\n✅ 测试脚本执行完毕。")
|