YOLO_Online 將深度學習最火的目標檢測做成在線服務
第一次接觸 YOLO 這個目標檢測項目的時候,我就在想,怎么樣能夠封裝一下讓普通人也能夠體驗深度學習最火的目標檢測項目,不需要關注技術細節,不需要裝很多軟件。只需要網頁就能體驗呢。
在踩了很多坑之后,終於實現了。
效果:
1.上傳文件
2.選擇了一張很多狗的圖片
3.YOLO 一下
技術實現
web 用了 Django 來做界面,就是上傳文件,保存文件這個功能。
YOLO 的實現用的是 keras-yolo3,直接導入yolo 官方的權重即可。
YOLO 和 web 的交互最后使用的是 socket。
坑1:
Django 中 Keras 初始化會有 bug,原計划是直接在 Django 里面用 keras,后來發現坑實在是太深了。
最后 Django 是負責拿文件,然后用 socket 把文件名傳給 yolo。
坑2:
說好的在線服務,為什么沒有上線呢?買了騰訊雲 1 CPU 2 G 內存,部署的時候發現 keras 根本起不來,直接被 Killed 。
解決,並沒有解決,因為買不起更好地服務器了,只好本地運行然后截圖了。
坑3:
YOLO 的識別是需要一定的時間的,做成 web 的服務,上傳完文件之后,並不能馬上識別出來,有一定的延遲。
相關教程:
TensorFlow + Keras 實戰 YOLO v3 目標檢測圖文並茂教程
https://zhuanlan.zhihu.com/p/36152438
YOLO QQ 群(求土豪贊助一波服務器啊)
群號:167122861
ObjectDetection:http://objectdetection.cn/