十行Python代码搞定图片中的物体检测

“Word is useless, show me the pic” -MR Lu

先看下原图：

图片表述的是一男一女在散步，后面有一辆车，现在来看下我们通过十行代码实现的效果：

我们可以看到，在这幅图中其实有三个“person”被识别出来，包括后面非常非常小的行人，还有一个“car”被识别出来，可以说模型能力基本达到了人眼的能力。现在就来介绍如何把这个识别结果用10行python代码实现。

代码

既然代码用得少，必然是站在了巨人的肩膀上去做事，需要安装大量第三方库。

1.需要首先用pip3安装以下库：

pip3 install numpy、scipy、opencv-python、pillow、matplotlib、h5py、keras、

注意：用到的是python3，有一些库在国内安装很慢，可以使用豆瓣源

pip3  install 第三方库名  -i  https://pypi.doubanio.com/simple/  --trusted-host pypi.doubanio.com

2.然后要安装用到的识别库，名字叫imageai：

pip3 install https://github.com/OlafenwaMoses/ImageAI/releases/download/2.0.1/imageai-2.0.1-py3-none-any.whl

3.接着要下载训练好的模型，我们直接使用现成的模型做预测，模型地址（145MB）：

https://github.com/OlafenwaMoses/ImageAI/releases/download/1.0/resnet50_coco_best_v2.0.1.h5

4.最后来看看代码：

from imageai.Detection import ObjectDetection
import osexecution_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 , "image3.jpg"), output_image_path=os.path.join(execution_path , "image3new.jpg"))for eachObject in detections:print(eachObject["name"] + " : " + eachObject["percentage_probability"] )print("--------------------------------")

clone代码地址：https://github.com/OlafenwaMoses/ImageAI

其中模型要跟代码放到统一路径下，image2.jpg是输入图片，image2new.jpg是输出图片。一共有效行数为10，不是你来打我，哈哈。运行这个文件就会生成标记好的图片。

是不是很简单，谢谢大家~

十行Python代码搞定图片中的物体检测相关推荐

About 一行Python代码搞定局域网内文件共享与传输
一行Python代码搞定局域网内文件共享与传输前提:环境中有Python安装,并配置环境变量. 1.共享目录:Windows下,在要共享文件目录下,shift+鼠标右键,在此处打开命令行窗口(CMD ...
python 判断图片中物体位置_Python 10行代码搞定图片中物体检测
使用Python, 如何编程实现图片中的物体检测[也叫对象检测/object detection]? 怎么知道图像中有哪些物体? 或许您想要计算图像中苹果的数量? 在这篇文章中,我将向您展示如何使用P ...
零基础如何用 15 行 Python 代码搞定网易云热门歌单？
健身.吃饭.敲代码:等车.撸猫.下午茶--若能佐以合适的音乐当"配餐",总是惬意非常.本文就将带你爬一爬网易云的那些热门歌单! 作者 | 上海小胖责编 | 仲培艺心情好或心情坏 ...
再见PDF提取收费！我用100行Python代码搞定！
点上方"菜鸟学Python",选择"星标" 第471篇原创干货,第一时间送达大家在日常的工作和学习过程中,都少不了与PDF文件打交道,很多的小伙伴都面临着将P ...
60行Python代码搞定，王者荣耀雷达图！
点击上方"菜鸟学Python",选择"星标"公众号超级无敌干货第一时间推给你!!! 现有以下表格,我需要画成第二张图中的雷达图(类似fifa足球.王者荣耀里面 ...
神操作！一行Python代码搞定一款游戏？给力！
来源:pypl编程榜一直以来Python长期霸占编程语言排行榜前三位,其简洁,功能强大的特性使越来越多的小伙伴开始学习Python .甚至K12的同学都开始学习Python 编程.新手入门的时候趣味 ...
10行Python代码搞定目标检测
from imageai.Detection import ObjectDetection #导入了 ImageAI 目标检测类 import os execution_path = os.getcw ...
怎样在python代码中输入π_鼠标自动点击、键盘自动输入？几行Python代码搞定
相信我,你以后一定会用到.实在不行,你先收藏,做个收藏家,等拿出来,就值钱了. 第一步:安装并导入相关模块只需要安装PyUserInput,如下: pip install PyUserInput 如 ...
再见PDF提取收费！我用100行Python代码搞定！去你的收费！
大家在日常的工作和学习过程中,都少不了与PDF文件打交道,很多的小伙伴都面临着将PDF文件中的文字.图片和表格数据提取出来的问题.能够对PDF文件中的文字.表格等数据进行编辑,网上现存的PDF提取的软 ...

十行Python代码搞定图片中的物体检测

代码

十行Python代码搞定图片中的物体检测相关推荐

最新文章

热门文章