自己搭建 AI 图像识别系统

Posted 2021-05-30 医学统计园

有这样一个应用场景：这里有一张图片，图片中有什么呢？

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

以上图像识别系统，可以在Ubuntu系统中实现（）。

下 载 代 码

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

tar -xzvf darknet.tar.gz

编 译 软 件

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

cd darknetmake

没有make命令，那么就装它

sudo apt install make

没有gcc命令，继续装它

sudo apt install gcc g++

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

sudo apt install g++

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

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

终于，顺利编译完成

运 行 图 片 识 别 系 统

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

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

data/dog.jpg 待识别图片

yolov3.weights 卷积网络权重

cfg/yolov3.cfg 网络配置

会出现卷积网络计算步骤

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

图像识别结果

你还可以继续尝试其他图片的识别，在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.