DAMO-YOLO-Colaboratory-Training-Sample

Demo.mp4

DAMO-YOLOをGoogle Colaboratory上で訓練しONNX形式のファイルをエクスポートするサンプルです。
以下の内容を含みます。

• データセット
• Colaboratory用スクリプト(環境設定、モデル訓練)
• ONNX推論サンプル

Requirement

• opencv-python 4.5.5.64 or later ※推論サンプルを実施する場合のみ
• onnxruntime 1.12.0 or later ※推論サンプルを実施する場合のみ

About annotation

Pascal VOC形式で出力したアノテーションデータを前提としています。
ただし、ノートブック内で更にMS COCO形式変換しています。

Usage

トレーニングはGoogle Colaboratory上で実施します。
[Open In Colab]リンクからノートブックを開き、以下の順に実行してください。

1. condaインストール(conda install)
   condacolab インストール後にセッション再起動でクラッシュしますが、インストールは成功しているため、以降の処理を続けて実行してください。
   After the installation, the session causes a crash, but the installation was successful, so please continue with the subsequent steps.
2. 必要パッケージインストール(install)
3. データセットダウンロード(Download Dataset)
4. Pascal VOC形式 を MS COCO形式へ変換(Convert Pascal VOC format to MS COCO format)
5. 学習データディレクトリ準備(Training data directory preparation)
6. DAMO-YOLO学習用コンフィグファイル生成
7. トレーニング
8. ONNXエクスポート
9. ダウンロード

Demo

トレーニング後のONNXの動作確認用デモは以下です。

python demo.py --model=damoyolo_tinynasL20_T.onnx --movie=fish.mp4

Author

高橋かずひと(https://twitter.com/KzhtTkhs)

License

DAMO-YOLO-Colaboratory-Training-Sample is under Apache-2.0 License.