hum_detector.py

# -*- coding: utf-8 -*-
"""
@author: tanma
"""

from imageai.Detection import VideoObjectDetection
import os

execution_path = os.getcwd()

detector = VideoObjectDetection()
detector.setModelTypeAsYOLOv3()
detector.setModelPath( os.path.join(execution_path , "yolo.h5"))
detector.loadModel()

custom_objects = detector.CustomObjects(person=True)

video_path = detector.detectCustomObjectsFromVideo(custom_objects=custom_objects,input_file_path=os.path.join( execution_path, "iitg.mp4"),
                                output_file_path=os.path.join(execution_path, "iitg_Detected_1")
                                , frames_per_second=12, log_progress=True)

print(video_path)