より高速な物体検出モデルを使いたい!!

かんたんに使える物体検出モデル、家電と同じで、どうせなら一番新しくていいやつを選びたいじゃないですか。
Yolov7〜高速物体検出シリーズの新しいバージョンでました、2022夏〜
Yolov5より120%高速とのこと。
使い方
学習済みモデルによる推論
python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source inference/images/horses.jpg
sourceには画像パス、もしくは画像フォルダ・パス、もしくは0(Webカメラ)を指定します。

ColabのGPUで0.057sで推論できました。
学習
python train.py --workers 8 --device 0 --batch-size 32 --data data/coco.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights '' --name yolov7 --hyp data/hyp.scratch.p5.yaml
モバイル・デバイスでの使用
リポジトリには、TorchScript、ONNX、CoreMLへのエクスポート・スクリプトがあり、かんたんにモバイル形式に変換して、アプリで使えます。
python models/export.py --weight yolov7.pt

僕がCoreMLに変換したモデルはこちら(NMSつき)
オリジナル・データでトレーニングして、便利なアプリを作りましょう!
🐣
フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com
Core MLやARKitを使ったアプリを作っています。
機械学習/AR関連の情報を発信しています。