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()