forked from mkocabas/VIBE
-
Notifications
You must be signed in to change notification settings - Fork 1
/
rt_demo.py
52 lines (34 loc) · 1.12 KB
/
rt_demo.py
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
import argparse
import cv2
from vibe.rt import RtVibe
def main(args):
rt_vibe = RtVibe()
if args.camera:
video = 0
elif args.vid_file:
video = args.vid_file
else:
video = 'sample_video.mp4'
rt_vibe.render = args.render
rt_vibe.sideview = args.sideview
cap = cv2.VideoCapture(video)
if not cap.isOpened():
print("Error opening video stream or file")
while cap.isOpened():
ret, frame = cap.read()
if ret:
result = rt_vibe(frame)
else:
break
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--vid_file', type=str,
help='input video path or youtube link')
parser.add_argument('--render', action='store_false',
help='render output')
parser.add_argument('--sideview', action='store_true',
help='render meshes from alternate viewpoint.')
parser.add_argument('--camera', action='store_true',
help='use camera as input')
args = parser.parse_args()
main(args)