YoloV5 マルチタスク ネットワークに基づいた分類ブランチを追加

---

序文

YoloV5 はターゲット検出タスクを実行しますが、画像内のターゲットを検出し、画像全体に従って画像を分類する必要がある場合は、Yolo-Detect 検出ヘッドの前に分類ブランチを追加して、バックボーンの状況を共有できます。以下に検出タスクと分類タスクを実装し、分類枝を追加する操作方法を詳しく説明します。

1. データの準備

以下の図に示すように、yolov5 label.txt ファイルの場合、1 行目にカテゴリ ラベル (0、1、2...) を追加するだけで、2 行目はターゲット ボックス bbox ラベルで始まります。

2. データ入力時のコードを変更します。

ラベル label に分類ブランチを追加する

3. ネットワークの変更

4. モデルの出力とラベルを接続して損失を計算します

5. 最終的な視覚化結果

元の yolo は 3 つの検出ヘッドを出力し、増加後は追加の分類ブランチを出力します。

要約する

yolo のバージョンも常に更新されているため、簡単に記録してください。詳細な説明がない場合は、同様の変更要件がある場合は、主な変更点をマークするだけです。おおよその考え方を参考にすることができます。丸ごとコピーすることはできません。