ケーススタディディーププラスティックとYOLOv5

ウルトラリティクスチーム

6分で読める

2022年12月12日

UltralyticsがAUVとYOLOv5を使用して効率的に水中での検知と清掃を行い、海洋プラスチック汚染にどのように取り組んでいるかをご覧ください。

プラスチックは海洋野生生物を窒息させている。毎分トラック2台分のプラスチックが海に投棄されており、これは年間1000万トン以上に相当する。ディーププラスチックの科学者たちは、この海洋プラスチックが「海洋環境、食の安全、人間の健康、エコツーリズム、気候変動への貢献」に対して社会的脅威をもたらすと述べている。

これに対抗するため、この研究者とエンジニアのチームは、コンピューター・ビジョンによって海洋のプラスチックをどのように除去できるかを研究している。

ディープ・プラスチックの研究者たちは、ディープ・ラーニング(深層学習)技術を使って、自律型水中探査機(AUV)を使い、光が透過する海面直下(表層)にあるプラスチックをスキャン、識別、定量化するアプローチを開発した。

「私たちの目標は、プラスチックの検出に使用できる、推論速度が非常に速い非常に小さなモデルを持つことでした」。
ジェイ・ロウ機械学習研究者

DeepPlasticチームは、YOLOv4と YOLOv5という2つの小型で精密なモデルを訓練し、リアルタイムの物体検出を可能にした。これらのモデルはDeepTrashデータセットで学習された:

  • 1900枚のトレーニング画像、637枚のテスト画像、637枚の検証画像(60、20、20分割)
  • カリフォルニア州タホ湖、サンフランシスコ湾、ボデガ湾で撮影されたフィールド写真。
  • Internet images (<20%) taken by scraping Google Images.
  • JAMSTEK JEDIデータセットからの深海画像

自律型水中ロボット(AUV)の仕組み

AUVは水中を移動するロボットである。深海まで自由に滑空し、海面に戻ってくることができる低速の乗り物である。AUVが水中でプラスチックを識別・回収するためには、ディープラーニング・モデルを搭載する必要がある。AUVは、水中でプラスチックを検出するために、3つの簡単なステップで展開することができる。

1.ディープラーニング・モデルをAUVに組み込む

2.海をスキャンする

3.プラスチックの特定

これまでのディープラーニング・モデルと海洋清掃の問題点

DeepPlasticチームは、YOLOv4やFaster R-CNNなどいくつかのディープラーニングモデルをAUVでテストした。しかし、研究者たちは、海洋洗浄を問題にするさまざまな課題に直面した。

限られた研究者へのアクセスがチームを停滞させた

チームにディープラーニングの専門家がいないため、研究者たちはディープラーニング・モデルを最大限に活用することができなかった。

推論速度が遅く、プラスチック検出が弱い

推論とは、AUVがどれだけ速くプラスチックを認識できるかということである。YOLOv4とFaster R-CNNでは、AUVはプラスチックを検出するのにそれほど効果的ではなく、水をきれいにする能力が損なわれた。

物体識別の精度が低い

YOLOv4とFaster R-CNNは、プラスチック識別の成功率が平均77%~80%しかなかった。

サンゴとプラスチックが混同される検出不良

Faster R-CNNを使用した場合、AUVによってプラスチックと識別されたサンゴは3~5%で、これは許容基準を下回った。

YOLOv5でパワーと精度が向上

YOLOv5に切り替えると、研究者たちはすぐに変化を実感した。精度は向上し、スピードは最大化され、YOLOv5のシンプルさによってチームの誰もが利用できるようになった。

より高速なR-CNNと比較して、平均20%高速な推論速度

93%の精度

YOLOv5のセットアップは1時間以内

アクセシビリティの向上により、研究者はYOLOv5を最大限に活用できるようになった

YOLOv5には、私たちがリポジトリ上で確立したシンプルなステップ・バイ・ステップのプロセスに基づいて、チームが簡単に作業できるようにするいくつかの側面がありました。

  • リポジトリのダウンロードは簡単だった
  • すべてのドキュメンテーションは、わかりやすく整理されていた。
  • 簡易モデル・トレーニング
  • 手動による結果チェック

より高い推論速度 最大限の海洋清掃効率

YOLOv5は、Faster RCNNよりも推論速度が20%速く、平均9ミリ秒で1枚の画像を処理した。その結果、AUVは浮遊プラスチックをより速いスピードで検出できるようになり、プラスチックの捕獲量とプロジェクト全体の効率が向上した。

高精度レートにおける精度の向上

精度は平均85%で、93%に達することもあった。これは以前のモデルで見られた平均77~80%から急上昇した。

より使いやすくなった研究者たち

YOLOv5のセットアップは、研究者たちにとってシームレスで簡単な体験でした。セットアップの全プロセスを通じて、ユーザーはAからZまでガイドされ、チームは1時間もかからずにYOLOv5を使い始めることができた。

YOLOv5をさまざまな水環境に適用できるようになった汎用性の向上

数日のうちに、補強なしの3000枚の画像からなる小さなデータセットを使って、研究グループは湖や川でAUVを訓練することができた。水が濁っているなどの悪条件にもかかわらず、YOLOv5で訓練されたAUVは、高い精度でプラスチックを検出・識別することができた。

「私たちは、高精度で非常に高速な物体検出アルゴリズムを探していました。私たちが作業する海洋環境は、過酷で荒い地形です。YOLOv5は、私たちが使用できる最高の物体検出モデルとして、すべての面で貢献してくれました。
「YOLOv5はセットアップも使い方も簡単で、一貫して望んでいた結果を出してくれるので、とても気に入っています。
「今後展開するモデルについては、間違いなくYOLOv5を第一候補として検討することになるだろう。
ゴータム・タタ機械学習研究者

DeepPlasticのリポジトリ発表された論文ビデオをご覧ください。

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

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

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