Agfdhyk

I'm trying to get video stream from camera and parse them. Here's my code:

import datetime

import os



import cv2

import face_recognition

from PIL import Image



import config

from helpers import get_known_faces, add_to_json, add_captured_face



print('Running...')

cap_vid = cv2.VideoCapture(config.stream_url)

while True:

    ret, frame = cap_vid.read()

    curr_face_encs = face_recognition.face_encodings(frame)

    curr_face_locs = face_recognition.face_locations(frame)

    face_names_in_DB, face_encs_in_DB = get_known_faces()

    if curr_face_encs:

        for idx, face in enumerate(curr_face_encs):

            res = face_recognition.compare_faces(face_encs_in_DB, face)

            if True in res:

                frame_data = {

                    'full_name': face_names_in_DB[res.index(True)],

                    'time': datetime.datetime.now().isoformat()

                }

                add_to_json(config.members_json_data, frame_data)

            else:

                top, right, bottom, left = curr_face_locs[idx]

                img = Image.fromarray(frame[top:bottom, left:right])

                img_name = '%s-%d.jpg' % (

                    datetime.datetime.now().isoformat().replace(':', 'c'), idx)

                img_path = os.path.join(config.stranger_dir, img_name)

                img.resize((40, 40)).save(img_path)

                add_captured_face(img_path,

                                datetime.datetime.now().isoformat())

When the config.stream_url is set to 0, i.e., when I'm reading from webcam, everything works perfectly. But when config.stream_url = rtsp://10.10.111.200/profile2/media.smp( an IP camera that I have full administrative access to and I can watch live video, say, via VLC player - what I'm trying to say is that url is working fine and I have access to it), it gives me the following error:

Traceback (most recent call last):

File "C:/Users/turalm01/Desktop/work/ITSFaceDetector/main.py", line 25, in <module>

    curr_face_encs = face_recognition.face_encodings(frame)

File "C:Usersturalm01AppDataLocalProgramsPythonPython36libsite-packagesface_recognitionapi.py", line 209, in face_encodings

    raw_landmarks = _raw_face_landmarks(face_image, known_face_locations, model="small")

File "C:Usersturalm01AppDataLocalProgramsPythonPython36libsite-packagesface_recognitionapi.py", line 153, in _raw_face_landmarks

    face_locations = _raw_face_locations(face_image)

File "C:Usersturalm01AppDataLocalProgramsPythonPython36libsite-packagesface_recognitionapi.py", line 102, in _raw_face_locations

    return face_detector(img, number_of_times_to_upsample)

RuntimeError: Unsupported image type, must be 8bit gray or RGB image.

I thought perhaps it wants me to convert each frame to RGB, so I tried converting each frame to RGB (but I'm not sure if this is how to do it), such as following:

import datetime

import os



import cv2

import face_recognition

from PIL import Image



import config

from helpers import get_known_faces, add_to_json, add_captured_face



print('Running...')

cap_vid = cv2.VideoCapture(config.stream_url)

while True:

    ret, frame = cap_vid.read()

    rgb_frame = frame[:, :, ::-1]

    curr_face_encs = face_recognition.face_encodings(rgb_frame)

    curr_face_locs = face_recognition.face_locations(rgb_frame)

    face_names_in_DB, face_encs_in_DB = get_known_faces()

    if curr_face_encs:

        for idx, face in enumerate(curr_face_encs):

            res = face_recognition.compare_faces(face_encs_in_DB, face)

            if True in res:

                frame_data = {

                    'full_name': face_names_in_DB[res.index(True)],

                    'time': datetime.datetime.now().isoformat()

                }

                add_to_json(config.members_json_data, frame_data)

            else:

                top, right, bottom, left = curr_face_locs[idx]

                img = Image.fromarray(rgb_frame[top:bottom, left:right])

                img_name = '%s-%d.jpg' % (

                    datetime.datetime.now().isoformat().replace(':', 'c'), idx)

                img_path = os.path.join(config.stranger_dir, img_name)

                img.resize((40, 40)).save(img_path)

                add_captured_face(img_path,

                                datetime.datetime.now().isoformat())

and this time it gave me the following error:

Traceback (most recent call last):

File "C:/Users/turalm01/Desktop/work/ITSFaceDetector/main.py", line 25, in <module>

    rgb_frame = frame[:, :, ::-1]

TypeError: 'NoneType' object is not subscriptable

Note: I want to mention a detail that perhaps you guys would like to know. In another Python module I'm getting live video stream from the same IP camera successfully but with a small bug, which is that the video that I'm receiving is the top right quarter of the whole frame.