ペットボトルごみの検出
ペットボトルごみの「キャップ,ラベル,ペットボトル」をYOLOで検出するコード
YOLOv5.ipynbまたはYOLOv8.ipynbのコード内で説明されています
また学習済みモデルはgoogle driveからダウンロードしてください
データセットとしてペットボトルが写った画像(267枚)を収集した。これに対してそれぞれの画像を縮小し45,135,225,315度回転させ水増しを行った。 これらをそれぞれ水増しなしのデータセットと水増しありのデータセットとして学習させた。
検出精度としてはYOLOv5xを使い水増しありのデータセットで学習させたモデルが最も優れていた。