-
Notifications
You must be signed in to change notification settings - Fork 2
/
camera6.py
28 lines (24 loc) · 1.06 KB
/
camera6.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
import cv2
import imutils
class VideoCamera(object):
def __init__(self):
# 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('rtsp://admin:[email protected]:554/PSIA/streaming/channels/601')
# 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):
while(self.video.isOpened()):
success, image = self.video.read()
image = imutils.resize(image, width=640)
print(image.shape)
if success==True:
# 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)
return jpeg.tostring()