-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathcamera.py
24 lines (22 loc) · 873 Bytes
/
camera.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
import cv2
import freenect
class VideoCamera(object):
def __init__(self):
pass
# Using OpenCV to capture from device 0. If you have trouble capturing
# from a webcam, comment the line below out and use a video file
# instead.
# self.video = cv2.VideoCapture(0)
# If you decide to use video.mp4, you must have this file in the folder
# as the main.py.
# self.video = cv2.VideoCapture('video.mp4')
# def __del__(self):
# self.video.release()
#
def get_frame(self):
image, success = freenect.sync_get_video()
# We are using Motion JPEG, but OpenCV defaults to capture raw images,
# so we must encode it into JPEG in order to correctly display the
# video stream.
ret, jpeg = cv2.imencode('.jpg', image[:, :, ::-1])
return jpeg.tobytes()