vlambda博客
学习文章列表

自己搭建 AI 图像识别系统


有这样一个应用场景:这里有一张图片,图片中有什么呢?


A:图片中有狗、单车和卡车

自己搭建 AI 图像识别系统


以上图像识别系统,可以在Ubuntu系统中实现()。


下 载 代 码

自己搭建 AI 图像识别系统

浏览器打开下载链接:http://www.yixuetongji.top/AIsystem/ , 下载完成之后,解压文件,是这个样的:

tar -xzvf darknet.tar.gz

自己搭建 AI 图像识别系统


编 译 软 件

自己搭建 AI 图像识别系统

进入到darknet/目录下,采用make命令进行编译(可以理解为windows下的安装)

cd darknetmake


没有make命令,那么就装它

自己搭建 AI 图像识别系统

sudo apt install make


没有gcc命令,继续装它

自己搭建 AI 图像识别系统

sudo apt install gcc g++


没有g++命令,好吧,装它

自己搭建 AI 图像识别系统

sudo apt install g++


依赖都已安装完成,再进行编译

#注意当前目录,要在darknet/ 目录下make


终于,顺利编译完成

自己搭建 AI 图像识别系统


运 行 图 片 识 别 系 统

自己搭建 AI 图像识别系统

先采用data/下自带的示例数据进行测试

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg


data/dog.jpg 待识别图片

yolov3.weights 卷积网络权重

cfg/yolov3.cfg 网络配置


会出现卷积网络计算步骤

自己搭建 AI 图像识别系统


计算结果在当前目录下的predictions.png

自己搭建 AI 图像识别系统


图像识别结果


你还可以继续尝试其他图片的识别,在data/下,比如

./darknet detect cfg/yolov3.cfg yolov3.weights data/kite.jpg

你自己DIY的任何图片,都可以去尝试。此外,借助CUDA和webcam,还可以实现动态识别,后续有时间会继续搭建。


参考文献

[1] Redmon J ,  Farhadi A . YOLOv3: An Incremental Improvement[J]. arXiv e-prints, 2018.