chenshijun
2019-06-05 3e0d6320a4acdb3c73e0efb31ee57fb1534237b2
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
import cv2
 
cap = cv2.VideoCapture('rtsp://admin:a1234567@192.168.1.188:554/H.264/ch1/main/av_stream')
if not cap.isOpened():
    print('not opened')
    exit()
 
# New cv2
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))   # float
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) # float
 
# Define the codec and create VideoWriter object
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out = cv2.VideoWriter('output/test/output-%s.avi' % 's', fourcc, 30, (width, height))
 
i = 0
while True:
    ret, frame = cap.read()
    i += 1
    if ret:
        print(i)
        # frame = cv2.flip(frame, 0)
        # cv2.imwrite('/home/basic/code/deep-head-pose/test/img_%s.png'%str(i), frame)
    if i > 100:
        # frame = cv2.flip(frame, 0)
        break
    out.write(frame)
 
out.release()
cap.release()
# cv2.destroyAllWindows()