良许Linux教程网 干货合集 如何通过10行代码完成图像识别

如何通过10行代码完成图像识别

ImageAI是一个 python 库,旨在使开发人员能够使用简单的几行代码构建具有包含深度学习和计算机视觉功能的应用程序和系统,下面将使用ImageAI为大家分享一下如何通过10行代码完成图像识别。

img

ImageAI安装工作

要使用ImageAI执行对象检测,您需要做的就是:

在计算机系统上安装Python 安装ImageAI及其依赖项 下载对象检测模型文件 运行示例代码(只有10行) 那么我们现在开始:

从官方Python语言网站下载并安装Python 3。 通过pip安装:TensorFlow,OpenCV, Keras, ImageAI

pip3 install tensorflow
pip3 install opencv-python
pip3 install keras
pip3 install imageai --upgrade

3)通过此文章中的链接下载用于对象检测的RetinaNet模型文件:

https://towardsdatascience.com/object-detection-with-10-lines-of-code-d6cb4d86f606

运行程序

太好了。我们现在已经安装了依赖项,可以编写第一个对象检测代码了。创建一个Python文件并给它起一个名字(例如,FirstDetection.py),然后将下面的代码写进去。将要检测的RetinaNet模型文件图像复制到包含python文件的文件夹中。

from imageai.Detection import ObjectDetection
import os
execution_path =os.getcwd()
detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage( input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))
for eachObject in detections:
print(eachObject["name"] , " : " ,
eachObject["percentage_probability"] )

需要注意的是,如果你在运行遇到这个错误:

ValueError: Unable to import backend : theano python mymodel.py

那么你可以尝试:

import osos.environ['KERAS_BACKEND'] = 'tensorflow'from
imageai.Detection import ObjectDetection

然后运行代码并等待结果打印在控制台中。一旦结果打印到控制台中,转到您的FirstDetection.py所在的文件夹,您将发现保存了一个新图像。看看下面的两个图像样本和检测后保存的新图像。

检测前: 十行代码完成图片识别,一起来看十行代码完成图片识别,一起来看 如何用10行代码完成目标检测 检测后: 十行代码完成图片识别,一起来看十行代码完成图片识别,一起来看 如何用10行代码完成目标检测 数据结果

我们可以看到程序会打印输出一些各个物体的概率数据:

person : 55.8402955532074
person : 53.21805477142334
person : 69.25139427185059
person : 76.41745209693909
bicycle : 80.30363917350769
person : 83.58567953109741
person : 89.06581997871399
truck : 63.10953497886658
person : 69.82483863830566
person : 77.11606621742249
bus : 98.00949096679688
truck : 84.02870297431946
car : 71.98476791381836

可以看出来程序可以对图片中的以下目标进行检测:

人,自行车,卡车,汽车,公交车。

大家可以直接将自己希望检测的照片放到程序里面运行看看效果。

原理解释

现在让我们解释一下10行代码是如何工作的。

from imageai.Detection import ObjectDetectionimport osexecution_path
= os.getcwd()

在上面的3行代码中,我们在第一行导入了ImageAI对象检测类,在第二行导入了python os类,并定义了一个变量来保存python文件、RetinaNet模型文件和图像所在的文件夹的路径。

detector = ObjectDetection()detector.setModelTypeAsRetinaNet()detector.setModelP
ath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))detector.loadModel()detections =
detector.detectObjectsFromImage(input_image=os.path.join(execution_pa
th , "image.jpg"), output_image_path=os.path.join(execution_path ,
"imagenew.jpg"))

在上面的代码中,我们定义对象检测类在第一线,将模型类型设置为RetinaNet在第二行,设置模型路径的路径在第三行RetinaNet模型,该模型加载到对象检测类在第四行,然后我们称为检测函数,解析输入图像的路径和输出图像路径在第五行。

for eachObject in detections: print(eachObject["name"] , " : "
, eachObject["percentage_probability"] )

在上面的代码中,我们在第一行迭代了detector.detectObjectsFromImage函数返回的所有结果,然后在第二行打印出图像中检测到的每个对象的模型名称和百分比概率。

以上就是良许教程网为各位朋友分享的Linux系统相关内容。想要了解更多Linux相关知识记得关注公众号“良许Linux”,或扫描下方二维码进行关注,更多干货等着你 !

img
本文由 良许Linux教程网 发布,可自由转载、引用,但需署名作者且注明文章出处。如转载至微信公众号,请在文末添加作者公众号二维码。
良许

作者: 良许

良许,世界500强企业Linux开发工程师,公众号【良许Linux】的作者,全网拥有超30W粉丝。个人标签:创业者,CSDN学院讲师,副业达人,流量玩家,摄影爱好者。
上一篇
下一篇

发表评论

联系我们

联系我们

公众号:良许Linux

在线咨询: QQ交谈

邮箱: yychuyu@163.com

关注微信
微信扫一扫关注我们

微信扫一扫关注我们

关注微博
返回顶部