YOLO_Online: 将深度学习最火的目标检测做成在线服务(附项目代码)
作者 | fendouai
编辑 | 磐石
出品 | 磐创AI技术团队
【磐创AI导读】:本文介绍了了如何将深度学习最火的目标检测做成在线服务。回复关键字“yolo_online”获得完整项目代码。欢迎大家点击上方蓝字关注我们的公众号:磐创AI。进入公众号通过下方文章精选系列文章了解更多keras相关项目。
第一次接触 YOLO 这个目标检测项目的时候,我就在想,怎么样能够封装一下让普通人也能够体验深度学习最火的目标检测项目,不需要关注技术细节,不需要装很多软件。只需要网页就能体验呢。
在踩了很多坑之后,终于实现了。
效果:
1.上传文件
2.选择了一张很多狗的图片
3.YOLO 一下
技术实现
1. web 用了 Django 来做界面,就是上传文件,保存文件这个功能。
2. YOLO 的实现用的是 keras-yolo3,直接导入yolo 官方的权重即可。
3. YOLO 和 web 的交互最后使用的是 socket。
tip0:
最理想的情况就是 YOLO 的模型和参数只加载一次,然后目标检测就很快。
这个比较容易实现,py 代码中先加载模型,再写处理逻辑。
tip1:
Django 中 Keras 初始化会有 bug,原计划是直接在 Django 里面用 keras,后来发现坑实在是太深了。
最后 Django 是负责拿文件,然后用 socket 把文件名传给 yolo。
tip2:
说好的在线服务,为什么没有上线呢?买了腾讯云 1 CPU 2 G 内存,部署的时候发现 keras 根本起不来,直接被 Killed 。
解决,只好本地运行然后截图了,这个服务对于硬件的要求还是挺高的。
tip3:
YOLO 的识别是需要一定的时间的,做成 web 的服务,上传完文件之后,并不能马上识别出来,有一定的延迟。
解决,正在尝试 web 端只负责接收图片,然后 GPU 服务器负责处理 YOLO 相关的任务,然后再把处理过的图片返回到 web 端。
相关教程:
TensorFlow + Keras 实战 YOLO v3 目标检测图文并茂教程(文末有惊喜)
YOLO QQ 群(求土豪赞助一波服务器啊)
群号:167122861
获取项目代码,关注微信公众号并发送关键字:yolo_online
写在最后:欢迎大家点击下方二维码关注我们的公众号,点击干货资源专栏或发送关键字“资源”获取更多资源推荐。关注我们的历史文章,和小编一起畅游在深度学习的世界中。
最后给大家送波福利,七月在线机器学习集训营课程免费试听了!!!
扫描下方海报二维码,领取免费试听课程,课程红包!
点击下方 | 阅读原文 | 了解更多
原创文章,作者:fendouai,如若转载,请注明出处:https://panchuang.net/2018/05/09/b1a6d27d9d/