-
Notifications
You must be signed in to change notification settings - Fork 3
/
record.py
36 lines (28 loc) · 941 Bytes
/
record.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
import pyrealsense2 as rs
import numpy as np
import cv2
# Configure depth and color streams
pipeline = rs.pipeline()
config = rs.config()
config.enable_stream(rs.stream.depth, 640, 480, rs.format.z16, 30)
config.enable_stream(rs.stream.color, 640, 480, rs.format.bgr8, 30)
config.enable_record_to_file('test.bag')
# Start streaming
pipeline.start(config)
e1 = cv2.getTickCount()
try:
while True:
# Wait for a coherent pair of frames: depth and color
# frames = pipeline.wait_for_frames()
# depth_frame = frames.get_depth_frame()
# color_frame = frames.get_color_frame()
# if not depth_frame or not color_frame:
# continue
e2 = cv2.getTickCount()
t = (e2 - e1) / cv2.getTickFrequency()
if t>30: # change it to record what length of video you are interested in
print("Done!")
break
finally:
# Stop streaming
pipeline.stop()