10行代码让你轻松搞定对象检测

摘要：十行代码轻松搞定对象检测？快来看看如何使用ImageAI。

计算机视觉是人工智能领域中最重要的一个分支。计算机视觉是一门能够识别和理解图像及其场景的计算机软件系统的科学。计算机视觉包括图像识别、对象检测、图像生成、图像超分辨率等多个方面。这其中有众多实际案例的就是对象检测，因此，对象检测可能是计算机视觉最有影响力的一个分支。本教程将简要介绍现代对象检测的概念、软件开发人员所面临的挑战、我们团队提供的解决方案以及高性能对象检测的代码教程。

对象检测是指计算机和软件系统在图像或场景中对目标对象进行定位和识别的能力。在实际应用中，目标检测已广泛用于人脸识别、车辆检测、行人计数、网络图像、安全系统和无人驾驶。并且，目前有很多种方法可用于对象检测。和其他任何计算机技术一样，对象检测的各种创造性和惊人的用途，来自于开发人员的不懈努力。

使用现代对象检测方法、以及基于这些方法构建新的应用程序并不是一个简单的任务。早期的对象检测大多使用经典算法，如计算机视觉库OpenCV中的算法。然而，在不同的环境中，这些经典算法的性能并没有达到我们所期望的性能。

2012年，深度学习有了新的突破，并被开发者广泛使用，出现了R-CNN、Fast-RCNN、Faster-RCNN、RetinaNet等现代高精度对象检测算法（方法），以及SSD和YOLO等快速高精度对象检测算法（方法）。要想使用这些基于深度学习（同时也基于机器学习）的算法进行对象检测，开发人员则需要掌握大量的数学和深度学习框架的相关知识。有数百万计的开发人员想要使用对象检测来创建新的产品。但是，由于深度学习技术理解和使用起来比较复杂，因此，这项技术很难掌握。

几个月前，我们就意识到了这一问题，我和John Olafenwa共同构建了ImageAI这个Python库，利用ImageAI，仅仅使用几行代码，开发人员就可以轻松地将最新的计算机视觉技术集成到现有的新应用程序中。

使用ImageAI进行对象检测的步骤如下：

1.在系统上安装Python。

2.安装ImageAI及其依赖项。

3.下载对象检测模型文件。

4.运行示例代码（10行）。

详细步骤如下：

1.从Python官网下载并安装Python 3。

2.使用pip命令安装以下依赖项：

(1)Tensorflow

命令:pip install tensorflow

(2)Numpy

命令:pip install numpy

(3)SciPy

命令:pip install scipy

(4)OpenCV

命令:pip install opencv-python

(5)Pillow

命令:pip install pillow

(6)Matplotlib

命令:pip install matplotlib

(7)H5py

命令:pip install h5py

(8)Keras

命令:pip install keras

(9)ImageAI

命令:pip installhttps://github.com/OlafenwaMoses/ImageAI/releases/download/2.0.1/imageai-2.0.1-py3-none-any.whl

3.下载用于对象检测的RetinaNet模型文件。

安装好了依赖项，现在就可以编写第一个对象检测代码了：创建一个Python文件并为其命名（如：FirstDetection.py），然后写入代码（如下图所示）。并将RetinaNet模型文件和要检测的图像复制到包含python文件的文件夹中。

FirstDetection.py

运行代码，并将结果打印到控制台，在FirstDetection.py所在的文件夹内，你就能找到保存的新图像。看看下面的2个图像样本和检测后保存的新图像。

对象检测前：

对象检测后：

控制台输出结果：

人 : 55.8402955532074

人 : 53.21805477142334

人: 69.25139427185059

人: 76.41745209693909

自行车: 80.30363917350769

人 : 83.58567953109741

人: 89.06581997871399

货车 : 63.10953497886658

人: 69.82483863830566

人: 77.11606621742249

公交车: 98.00949096679688

货车 : 84.02870297431946

小汽车: 71.98476791381836

控制台输出结果：

人: 71.10445499420166

人: 59.28672552108765

人: 59.61582064628601

人: 75.86382627487183

摩托车: 60.1050078868866

公交车: 99.39600229263306

小汽车 : 74.05484318733215

人: 67.31776595115662

人 : 63.53200078010559

人: 78.2265305519104

人: 62.880998849868774

人: 72.93365597724915

人: 60.01397967338562

人: 81.05944991111755

摩托车: 50.591760873794556

摩托车: 58.719027042388916

人: 71.69321775436401

自行车: 91.86570048332214

摩托车: 85.38855314254761

现在，我们来看一下这10行代码是如何运行的。

如上图所示，第一行导入了ImageAI对象检测类，第二行导入了python os类，第三行定义了一个变量来保存python文件、RetinaNet模型文件和图像所在的文件夹的路径。

在上图中的5行代码中，第一行中定义了对象检测类，第二行将模型类型设置为RetinaNet，第三行将模型路径设置为RetinaNet模型的路径，第四行将模型加载到对象检测类，第五行调用检测函数，并在输入和输出图像路径中分别进行解析。

在上面的两行代码中，第一行将detector.detectObjectsFromImage函数返回的所有结果进行迭代，第二行打印出模型在每个图像上检测出来的对象名称和概率。

ImageAI支持对象检测过程的定制，其中之一就是能够提取图像中检测到的每个对象。如下所示，通过简单地将extra_detected_objects = True设置为detectObjectsFromImage函数的参数，对象检测模型将为检测到的对象创建创建一个文件夹，提取每个图像，并将每个图像都保存到新创建的文件夹中，并返回一个包含每个对象的数组。