こんにちは、システムズナカシマの松浦です。
今回は学習について解説します。
これまで作成したデータセットをもとにAIモデルを学習し、学習したAIモデルで推論を試す手順について解説します。
本記事をお読みになる前に、アカウントをお持ちでない方（サインアップ）、ワークスペース・プロジェクトの準備がお済みでない方は以下の記事をご確認ください。
また、roboflowで何ができるか、概要を知りたい方は「はじめてのroboflow」の記事をご参照ください。
学習
前回までの記事で作成したデータセットを利用しAIモデルの学習を行います。
画面左に表示されているメニューの「Versions」メニューを選択すると、前回の記事で作成したデータセットが表示されます。
その横に表示される「Train with Roboflow」ボタンを押下します。
モデルの選択画面が表示されるので学習するモデルを選択します。ここでは「Roboflow 3.0」モデルを選択し「Continue」ボタンを押下します。
| モデルの種類 | 説明 |
|---|---|
| Roboflow 3.0 | 推奨。Roboflowオリジナル。ホスティッドインファレンス、ローカル、エッジデバイスでの実行が可能 |
| YOLOv11 | ホスティッドインファレンス、ローカル実行が可能 |
| YOLO-NAS | 推論スピードは速くない。ホスティッドインファレンス、ローカル実行が可能 |
「Roboflow 3.0」モデルを選択すると以下の画面が表示されます。用途に応じでモデルの種類を選択します。
| モデルの種類 | 推論スピード | 学習時間 | 精度 |
|---|---|---|---|
| Fast | 速い 
|
短い | 低め  |
| Accurate | 遅い 
|
長い | 高め  |
「Continue」ボタンを押下すると「Checkpoint」を選択する画面が表示されるので、「Train from Public Checkpoint」を選択し「Start Training」ボタンを押下します。
[ワンポイント]学習データを追加し次回学習を行う場合は「Train from Previous Checkpoint」を選択し学習を行います。
学習が実行されたことを表す画面が表示されます。学習にどのくらいの時間がかかるか目安となる時間が表示されます。
学習の状況はリアルタイムに画面上に表示されます。
学習完了
ログインに利用しているメールアドレスに以下の学習完了をお知らせるメールが届きます。
また、「Versions」メニューの学習を行なったバージョンに
チェックマークが表示されます。画面右側には、「mAP」、「Precision」、「Recall」といった学習結果の指標が表示されます。
| 項目 | 説明 | 備考 |
|---|---|---|
| mAP（平均適合率の平均） | クラス毎の適合率の全体の平均値。 | |
| Precision（適合率） | 推論した結果がどれだけ正確だったか。 | そもそも推論されなかったものが考慮されない。 |
| Recall（再現率） | 推論すべきものをどれだけ正確に推論できたか。 | 推論すべきものをどれだけ取りこぼしなく推論できたかを評価できる。 |
学習結果の評価
上記、「mAP」、「Precision」、「Recall」以外に「Detailed Model Evaluation」にて混同行列(Confusion Matrix)を確認することができます。
混同行列(Confusion Matrix)では、モデルの予測結果とアノテーションした際のクラスとの間の一致や不一致を表形式で確認することができます。
混同行列の利用方法について少し解説します。
表の横軸（PREDICTION）はモデルが推論した結果、縦軸（GROUND TRUTH）は、アノテーション時に設定したクラス（正解）を表します。
よって、表の対角線上の数値は正しくモデルが推論できたことを表しており、それ以外の数値に注意する必要があります。
以下では、表の一番右にある列（False Negative）の「ok_ebi」行の確認を行なっています。セル数値を押下すると実際に「ok_ebi」が検出できなかった画像が確認できます。
このモデルの精度を上げるのにどういった画像を次回収集すべきかの判断材料になりますね!! 
モデル実行
その他、簡単に学習したモデルでの推論を確認することができます。「Try This Model」を押下すると画像を選択し推論結果を確認することができます。
「Try on mobile」のQRコードをお手持ちのスマートフォンで読み取るとスマートフォンのカメラを利用し確認することができるので是非お試しください!!
最後に
いかがでしたでしょうか？
今回は、モデルの学習方法について説明しました。自分でGPUを用いた学習環境の準備が不要なので簡単に学習することができます。
次回は「第6回 roboflow：推論結果の確認」にて学習したモデルによる推論結果の確認方法について説明します。
※記事一覧
システムズナカシマでは、roboflowに関する個別相談・無料デモを行っております。
是非お気軽にご相談ください。