|
|
from auto_media_publisher.modifiers.video_matting.robust_video_matting_converter import RobustVideoMattingConverter
|
import time
|
if __name__ == '__main__':
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
parser.add_argument('--input', type=str, required=True)
|
parser.add_argument('--bg_video', type=str)
|
parser.add_argument('--bg_image', type=str)
|
parser.add_argument('--output_video', type=str)
|
parser.add_argument('--output_alpha', type=str)
|
parser.add_argument('--output_fore', type=str)
|
|
args = parser.parse_args()
|
|
time_start = time.time()
|
|
converter = RobustVideoMattingConverter()
|
|
print(f"model loaded .. {time.time()-time_start}")
|
time_start = time.time()
|
|
converter.video_matting(
|
input=args.input,
|
bg_video=args.bg_video,
|
bg_image=args.bg_image,
|
output_video=args.output_video,
|
output_alpha=args.output_alpha,
|
output_fore=args.output_fore,
|
)
|
print(f"video matting finished .. {time.time()-time_start}")
|
# conda activate mediapublish
|
# f:
|
# cd automedia
|
#python -m test.test_video_matting --input ./temp/buhui.mp4 --bg_video ./temp/ocean.mp4 --output_video ./temp/buhui_ocean-36.mp4
|
