|
|
import os |
|
|
import sys |
|
|
import argparse |
|
|
|
|
|
VIDEO_FILE_DIR = "/home/xiuying/Code/xywang/test_videos" |
|
|
|
|
|
|
|
|
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✅ 测试脚本执行完毕。") |
|
|
|
