自己搭建 AI 图像识别系统
有这样一个应用场景:这里有一张图片,图片中有什么呢?
A:图片中有狗、单车和卡车
以上图像识别系统,可以在Ubuntu系统中实现()。
下 载 代 码
浏览器打开下载链接:http://www.yixuetongji.top/AIsystem/ , 下载完成之后,解压文件,是这个样的:
tar -xzvf darknet.tar.gz
编 译 软 件
进入到darknet/目录下,采用make命令进行编译(可以理解为windows下的安装)
cd darknet
make
没有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.