Flaskを用いてyolo5モデルを公開するリポジトリ

Flaskを用いてyolo5モデルを公開するにあたり、以下のリポジトリが参考になりました。

 

github.com

 

画像のアップロードフォームを含むシンプルなWebアプリと、RestAPIのサンプルなどが提供されています。

 

カスタムモデルのロード

load関数の引数を以下のように変更することで、独自に学習したモデルなどをローカルからロードすることができました。

 

model = torch.hub.load("ultralytics/yolov5", "custom", path='model/best.pt', force_reload=True, autoshape=True) 

 

yolov5-flask/webapp.py at f1ee21bc3b14cdf1a1ea3bc95111a8db31172216 · robmarkcole/yolov5-flask
Official implementation at https://github.com/ultralytics/yolov5/tree/master/utils/flask_rest_api - robmarkcole/yolov5-flask
GitHub

 

pathの値は、適宜変更してください。force_reloadやautoshareはオプションで、私の環境では未指定でも動作しました。

 

以上、参考になりましたら幸いです。