| | |
|---|
| | | |
|---|
| | | |
|---|
| | | from auto_media_publisher.modifiers.video_matting.video_matting_converter import VideoMattingConverter |
|---|
| | | 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_jpg', 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) |
|---|
| | |
|---|
| | | |
|---|
| | | time_start = time.time() |
|---|
| | | |
|---|
| | | converter = VideoMattingConverter() |
|---|
| | | 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_jpg=args.bg_jpg, |
|---|
| | | bg_image=args.bg_image, |
|---|
| | | output_video=args.output_video, |
|---|
| | | output_alpha=args.output_alpha, |
|---|
| | | output_fore=args.output_fore, |
|---|
