Python实现ImageAI ，视频中目标检测10行代码

最新推荐文章于 2024-08-08 07:49:23 发布

人工智能之浪潮

最新推荐文章于 2024-08-08 07:49:23 发布

阅读量2.2k

点赞数

分类专栏：计算机视觉算法深度学习&&机器学习

本文链接：https://blog.csdn.net/mago2015/article/details/85336930

版权

计算机视觉算法同时被 2 个专栏收录

25 篇文章 8 订阅

订阅专栏

深度学习&&机器学习

14 篇文章 1 订阅

订阅专栏

Python实现Imageai，视频中目标检测10行代码

ImageAI 提供方便，灵活和强大的方法来对视频进行对象检测和跟踪。目前仅支持当前最先进的 RetinaNet 算法进行对象检测和跟踪

from imageai.Detection import VideoObjectDetection
import os
execution_path = os.getcwd()
detector = VideoObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath(os.path.join(execution_path, "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
# custom_objects = detector.CustomObjects(person=True, bicycle=True, motorcycle=True)
video_path = detector.detectObjectsFromVideo(input_file_path=os.path.join(execution_path, "saveVideo.avi"), output_file_path=os.path.join(execution_path, "traffic_detected"), frames_per_second=20, log_progress=True)
print(video_path)

效果图：