深度学习和目标检测系列教程 2-300：小试牛刀，使用 ImageAI 进行对象检测

@Author：Runsen

对象检测是一种属于更广泛的计算机视觉领域的技术。它处理识别和跟踪图像和视频中存在的对象。目标检测有多种应用，如人脸检测、车辆检测、行人计数、自动驾驶汽车、安全系统等。ImageAI提供了非常方便和强大的方法来对图像进行对象检测并从图像中提取每个对象。

ImageAI

ImageAI 包含几乎所有最先进的深度学习算法的 Python 实现，如RetinaNet、YOLOv3和 TinyYOLOv3。

ImageAI 使用了对象检测、视频检测和对象跟踪 API，无需访问网络即可调用。ImageAI 使用预先训练的模型并且可以轻松定制。

ImageAI 中的ObjectDetectionImageAI 库的类包含使用预训练模型对任何图像或图像集执行对象检测的函数。借助 ImageAI，可以检测和识别 80 种不同的常见日常物品。

具体官方教程：https://imageai.readthedocs.io/en/latest/detection/index.html

下载链接：https://github.com/OlafenwaMoses/ImageAI/releases/download/2.0.2/imageai-2.0.2-py3-none-any.whl

安装ImageAI 成功后，下载包含将用于对象检测的分类模型的TinyYOLOv3模型文件。

下载地址：https://github.com/OlafenwaMoses/ImageAI/releases/download/1.0/yolo-tiny.h5

下面，我们小试牛刀，使用 ImageAI 进行对象检测

我们先创建必要的文件夹。

Object-detection：根文件夹
models：存储预先训练的模型
input : 存储我们要执行对象检测的图像文件
output：存储检测到对象的图像文件

创建文件夹后，Object-detection文件夹应具有以下子文件夹：

├── input
├── models
└── output

下面创建一个新文件detector.py.，从 ImageAI 库导入ObjectDetection类。

from imageai.Detection import ObjectDetection

接下来是创建 class 的实例ObjectDetection，如下所示：

detector = ObjectDetection()

指定输入图像、输出图像和模型的路径。

model_path = "./models/yolo-tiny.h5"
input_path = "./input/image.jpg"
output_path = "./output/newimage.jpg"

实例化ObjectDetection类后，我们现在可以从类中调用各种函数。该类包含以下功能调用预先训练模式：setModelTypeAsRetinaNet()，setModelTypeAsYOLOv3()，和setModelTypeAsTinyYOLOv3()。

我使用预训练TinyYOLOv3模型，因此我们将使用该setModelTypeAsTinyYOLOv3()函数加载我们的模型。

接下来，将调用函数setModelPath()。此函数接受一个包含预训练模型路径的字符串：

detector.setModelPath(model_path)

从detector实例调用函数loadModel()。它使用setModelPath()类方法从上面指定的路径加载模型

detector.loadModel()

要检测图像中的对象，我们需要detectObjectsFromImage使用detector创建的对象。

下面是完整代码

from imageai.Detection import ObjectDetectiondetector = ObjectDetection()model_path = "./models/yolo-tiny.h5"
input_path = "./input/img.png"
output_path = "./output/img.png"detector.setModelTypeAsTinyYOLOv3()
detector.setModelPath(model_path)
detector.loadModel()
detection = detector.detectObjectsFromImage(input_image=input_path, output_image_path=output_path)for eachItem in detection:print(eachItem["name"] , " : ", eachItem["percentage_probability"])

深度学习和目标检测系列教程 2-300：小试牛刀，使用 ImageAI 进行对象检测相关推荐

深度学习之对象检测_深度学习时代您应该阅读的12篇文章，以了解对象检测
深度学习之对象检测前言 (Foreword) As the second article in the "Papers You Should Read" series, we a ...
综述 | 基于深度学习的目标检测算法
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达本文转自:计算机视觉life 导读:目标检测(Object Det ...
病虫害模型算法_基于深度学习的目标检测算法综述
sigai 基于深度学习的目标检测算法综述导言目标检测的任务是找出图像中所有感兴趣的目标(物体),确定它们的位置和大小,是机器视觉领域的核心问题之一.由于各类物体有不同的外观,形状,姿态,加上成像 ...
基于深度学习的目标检测的研究进展2
普通的深度学习监督算法主要是用来做分类,如图1(1)所示,分类的目标是要识别出图中所示是一只猫.而在ILSVRC(ImageNet Large Scale Visual Recognition Cha ...
基于深度学习的目标检测研究进展
前言开始本文内容之前,我们先来看一下上边左侧的这张图,从图中你看到了什么物体?他们在什么位置?这还不简单,图中有一个猫和一个人,具体的位置就是上图右侧图像两个边框(bounding-box)所在的位 ...
AI综述专栏 | 基于深度学习的目标检测算法综述
https://www.toutiao.com/a6685618909275488780/ 2019-04-30 17:35:53 关注微信公众号:人工智能前沿讲习, 重磅干货,第一时间送达 AI综述 ...
基于深度学习的目标检测算法综述(从R-CNN到Mask R-CNN)
深度学习目标检测模型全面综述:Faster R-CNN.R-FCN和SSD 从RCNN到SSD,这应该是最全的一份目标检测算法盘点基于深度学习的目标检测算法综述(一) 基于深度学习的目标检测算法综述 ...
【深度学习】基于深度学习的目标检测研究进展
原文出处:http://chuansong.me/n/353443351445 开始本文内容之前,我们先来看一下上边左侧的这张图,从图中你看到了什么物体?他们在什么位置?这还不简单,图中有一个猫和一个 ...
基于深度学习的目标检测方法综述
引言现有的深度学习的目标检测方法,可以大致分为两类:一.基于候选区域的目标检测方法:二.基于回归的目标检测方法.依据方法的提出时间,可以构建出如下时间线: 2014 CVPR R-CNN[1] 20 ...

深度学习和目标检测系列教程 2-300：小试牛刀，使用 ImageAI 进行对象检测

ImageAI

深度学习和目标检测系列教程 2-300：小试牛刀，使用 ImageAI 进行对象检测相关推荐

最新文章

热门文章