Google ColabのUltralytics YOLO11による画像セグメンテーション

アビラミ・ヴィナ

4 min read

2024年12月30日

Ultralytics YOLO11を画像セグメンテーションに効果的に使用し、Google Colab上の自動車部品データセットを活用して、シームレスなトレーニングとテストを行う方法をご紹介します。

最新のUltralytics YOLO11のようなUltralytics YOLOモデルは、物体検出、画像分類、インスタンス分割など、さまざまなコンピュータビジョンタスクをサポートします。これらのタスクはそれぞれ、人間の視覚の特定の側面を再現することを目的としており、機械が周囲の世界を見て解釈できるようにします。 

例えば、美術の授業で生徒が鉛筆を手に取り、対象物の輪郭を描いている様子を考えてみよう。その背後では、脳がセグメンテーション(対象物を背景やその他の要素から区別すること)を行っているのだ。画像セグメンテーションは、人工知能(AI)を使用して同様の目標を達成し、視覚データを機械が理解できるように意味のある部分に分解する。この技術は、さまざまな業界のさまざまなアプリケーションで使用することができる。 

__wf_reserved_inherit
図1. Ultralytics YOLO11を使用して画像内のオブジェクトをセグメント化。

実用的な例として、自動車部品のセグメンテーションがある。自動車の特定の部品を識別し、分類することで、画像セグメンテーションは、自動車製造、修理、eコマースカタログのような業界のプロセスを合理化することができる。

この記事では、Ultralytics YOLO11、Google Colab、Roboflow Carparts Segmentationデータセットを使用して、車の部品を正確に識別し、セグメント化するソリューションを構築する方法を探ります。

ウルトラリティクスYOLO11は使いやすい

Ultralytics YOLO11は、80種類のオブジェクトクラスをカバーするCOCOデータセットで学習済みのモデルとして提供されています。しかし、車の部品のセグメンテーションのような特定のアプリケーションのために、モデルはデータセットとユースケースに適したカスタムトレーニングが可能です。この柔軟性により、YOLO11は汎用的なタスクでも高度に専門化されたタスクでも優れた性能を発揮します。

カスタムトレーニングでは、事前にトレーニングされたYOLO11モデルを使用し、新しいデータセット上で微調整を行います。タスクに特化したラベル付き例を提供することで、モデルはプロジェクト固有のオブジェクトを認識し、セグメント化することを学習します。カスタム・トレーニングは、事前にトレーニングされた一般的な重みに頼るよりも高い精度と関連性を保証します。

カスタムトレーニングのためのYOLO11のセットアップは簡単です。最小限のセットアップで、モデルとデータセットをロードし、トレーニングを開始し、プロセス中に損失や精度などのメトリクスをモニターすることができます。YOLO11には、検証と評価のためのツールも組み込まれており、モデルのパフォーマンスを簡単に評価することができます。 

Google ColabでUltralytics YOLO11を動かす

YOLO11をカスタム・トレーニングする場合、環境を設定するためのいくつかの選択肢がある。最も身近で便利な選択肢の一つがGoogle Colabです。YOLO11のトレーニングにGoogle Colabを使う利点は以下の通りです:

  • リソースへの無料アクセス:Google ColabはGPU(Graphics Processing Unit) とTPU(Tensor Processing Unit)を提供しており、高価なハードウェアを使用せずにYOLO11をトレーニングすることができます。
    ‍。
  • コラボレーション環境:Google Colab では、ノートブックを共有し、Google Drive に作業を保存し、簡単なコラボレーションとバージョン追跡によってチームワークを簡素化できます。
    ‍。
  • プリインストールされたライブラリ:PyTorchやTensorFlowなどのプリインストールツールにより、Google Colabはセットアッププロセスを簡素化し、迅速な開始を支援します。
  • クラウドとの統合:Google Drive、GitHub、その他のクラウドソースからデータセットを簡単に読み込むことができ、データの準備と保存を簡素化できます。
__wf_reserved_inherit
図2.Google Colab YOLO11ノートブック。

Ultralyticsでは、YOLO11トレーニング用に特別に設定済みのGoogle Colabノートブックも提供しています。このノートブックには、モデルのトレーニングからパフォーマンス評価まで、必要なものがすべて含まれており、プロセスをわかりやすく簡単に行うことができます。複雑な設定手順を気にすることなく、特定のニーズに合わせてモデルを微調整することに集中できます。

Roboflow Carparts Segmentation Datasetの概要

トレーニング環境を決定したら、次のステップは、データを収集するか、車の部品をセグメンテーションするのに適したデータセットを選択することです。Roboflow Universeで公開されているRoboflowCarparts Segmentation Datasetは、コンピュータビジョンモデルの構築、トレーニング、デプロイのためのツールを提供するプラットフォームであるRoboflowによって管理されています。このデータセットには、3,156枚の学習用画像、401枚の検証用画像、276枚のテスト用画像が含まれており、バンパー、ドア、ミラー、ホイールなどの自動車部品に関する高品質なアノテーションが付与されています。

通常、Roboflow Universeからデータセットをダウンロードし、Google Collab上でトレーニングするために手動で設定する必要があります。しかし、Ultralytics Pythonパッケージは、シームレスな統合と設定済みのツールを提供することで、このプロセスを簡素化します。

__wf_reserved_inherit
図3.自動車部品のセグメンテーションデータセットの例。

Ultralyticsでは、データセットのパス、クラスラベル、その他の学習パラメータを含む設定済みのYAMLファイルを通して、データセットをすぐに使用することができます。このファイルには、データセットのパス、クラスラベル、その他のトレーニングパラメータが含まれています。また、データセットは専用のトレーニングセット、検証セット、テストセットで構成されているため、進捗状況の確認や性能評価が容易になります。

Roboflow Carparts Segmentation DatasetとUltralytics YOLO11が提供するツールを活用することで、Google Colabのようなプラットフォーム上で効率的にセグメンテーションモデルを構築するためのシームレスなワークフローを手に入れることができます。このアプローチにより、セットアップ時間を短縮し、実世界でのアプリケーションのためのモデルの改良に集中することができます。

自動車部品セグメンテーションの実際の応用例

自動車部品のセグメンテーションは、さまざまな業界で実用的に使用されている。例えば、修理工場では、損傷した部品をすばやく識別して分類することで、修理プロセスをより迅速かつ効率的に行うことができる。同様に、保険業界では、セグメンテーションモデルは、損傷した車両の画像を分析して、影響を受けた部品を特定することで、クレーム査定を自動化することができる。これにより、保険金請求プロセスを迅速化し、ミスを減らし、保険会社と顧客の双方にとって時間を節約することができる。

__wf_reserved_inherit
図4.YOLOを使った自動車部品の分割。

製造に関しては、セグメンテーションは、自動車部品の欠陥を検査し、一貫性を確保し、無駄を省くことによって、品質管理をサポートします。これらのアプリケーションは、自動車部品のセグメンテーションが、プロセスをより安全、迅速、正確にすることで、産業をどのように変革できるかを示しています。

ステップバイステップガイド:Google ColabでYOLO11を使う 

さて、すべての詳細をカバーしたところで、いよいよすべてをまとめましょう。まずは、YouTubeのビデオで、自動車部品セグメンテーションのためのYOLO11モデルのセットアップ、トレーニング、検証の全プロセスをご覧ください。

ここでは、その手順を簡単に紹介しよう:

  • Google Colabで環境をセットアップします:GPUサポートを有効にし、Ultralytics Pythonパッケージをインストールしてモデルトレーニングの準備をする。
  • YOLO11モデルをロードします:時間を節約し、自動車部品のセグメンテーションに既存の特徴を活用するために、事前にトレーニングされたYOLO11セグメンテーションモデルから始めます。
  • データセットを使ってモデルをトレーニングする:データセットを使ってモデルをトレーニングする:トレーニング中に "carparts-seg.yaml "ファイルを使い、Roboflow Carparts Segmentation Datasetを自動的にダウンロード、設定、使用する。エポック、画像サイズ、バッチサイズなどのパラメータを調整し、モデルを微調整する。
  • トレーニングの進捗を監視します:セグメンテーションロスや平均平均精度(mAP)などの主要なパフォーマ ンス指標を追跡し、モデルが期待通りに向上していることを確認する。
    ‍ 以下のようになる。
  • モデルの検証とデプロイ検証セットで学習済みモデルをテストして精度を確認し、品質管理や保険金請求処理などの実世界のアプリケーション用にエクスポートします。

自動車部品のセグメンテーションにYOLO11を使用するメリット

YOLO11は、自動車部品のセグメンテーションのための信頼できる効率的なツールであり、様々な実際の用途に理想的な利点を提供します。主な利点は以下の通りです:

  • スピードと効率:YOLO11は、高い精度を維持しながら画像を高速処理するため、品質管理や自律走行車などのリアルタイムタスクに適している。
  • 高い精度:このモデルは、1つの画像内の複数のオブジェクトを検出し、セグメンテーションすることに優れており、自動車部品の正確な識別を保証する。
  • 拡張性:YOLO11は、大規模なデータセットと複雑なセグメンテーション・タスクを処理できるため、産業用アプリケーションに適したスケーラビリティを備えている。
  • 複数の 統合:Ultralyticsは、Google Colab、Ultralytics Hub、その他の一般的なツールなどのプラットフォームとの統合をサポートしており、開発者の柔軟性とアクセシビリティを高めています。

Google CollabでYOLO11と作業するためのヒント

Google Colabは機械学習ワークフローをより簡単に扱うことができるが、初めて使う場合は慣れるまで少し時間がかかるかもしれない。クラウドベースのセットアップ、ランタイムの設定、セッションの制限をナビゲートするのは、最初は厄介に感じるかもしれないが、物事をよりスムーズにするヒントがいくつかある。

以下は、留意すべきいくつかの点である:

  • トレーニングをスピードアップするために、ランタイム設定でGPUアクセラレーションを有効にすることから始めましょう。
  • Colabはクラウド上で動作するため、データセットやリポジトリなどのリソースにアクセスする際には安定したインターネット接続が必要です。
  • ファイルやデータセットをGoogle DriveやGitHubで整理し、Colab内で簡単に読み込んで管理できるようにします。
  • Colabの無料版でメモリ制限に遭遇した場合は、トレーニング中の画像サイズまたはバッチサイズを小さくしてみてください。
  • Colabセッションには時間制限があり、進捗を失いたくないので、定期的にモデルと結果を保存することを忘れないでください。 

YOLO11でさらなる成果を

Ultralytics YOLO11は、Google ColabのようなプラットフォームやRoboflow Carparts Segmentationデータセットのようなデータセットと組み合わせることで、画像セグメンテーションを簡単かつ身近なものにします。直感的なツール、事前にトレーニングされたモデル、簡単なセットアップにより、YOLO11は高度なコンピュータビジョンのタスクに簡単に飛び込むことができます。 

自動車の安全性の向上、製造の最適化、革新的なAIアプリケーションの構築など、この組み合わせは成功のためのツールを提供します。Ultralytics YOLO11を使えば、単にモデルを構築するだけでなく、現実の世界でよりスマートで効率的なソリューションへの道を開くことができます。

さらに詳しく知りたい方は、GitHubリポジトリをチェックし、私たちのコミュニティに参加してください。自動運転車や 農業用コンピュータビジョンにおけるAIアプリケーションについては、ソリューションページをご覧ください。🚀

AIの未来
を一緒に作りましょう!

機械学習の未来への旅を始めよう

無料で始める
クリップボードにコピーされたリンク