YOLO V5 をGoogle Colabで学習する方法

Let’s play with AI.

こんにちは。

AI coordinator管理人の清水秀樹です。

前回に引き続き今回はYOLO V5の学習モデルをGoogle Colabで作る方法をご紹介します。

Google ColabはGoogleアカウントがあれば誰でも無料に使えるサービスで、AI体験もすぐできる優れた環境になっています。

無料のため利用制限はありますが、高価なGPU搭載PCなんかを購入しなくても学習モデルが作れたりするので非常に便利ですね。

教育用にも使えるのでオススメです。

しかもOSに依存しない。

素晴らしい。

それでは早速準備していきましょう。

まずはGitHubリポジトリからソースをダウンロードしてください。

学習データは公開データセットを使用します。

Roboflow公開データセット

色々な学習データがありますが、今回はマスクのデータセットを使用します。

で、文章で説明するよりも動画で説明したほうが分かりやすいかと思いますので、以下のyoutube動画を参照ください。

動画の通りになぞって頂ければ、学習モデル作成から、その利用まで無事出来るかと思います。

考察

高価なPCを購入しなくても簡単に実行できる環境が整いつつあるのは非常に嬉しい限りですね。

参入障壁もグッと下がる可能性を秘めているのかなと。

私も引き続き頑張っていきたいと思います。

それではまた。

この記事の内容について、相談したい方へ

技術選定・実装の進め方・組織での導入など、AI/IoT全般のご相談を受け付けています。

→ 無料相談会（30分・オンライン）

About The Author

Hideki

東京大学発AIスタートアップでロボット開発室室長・画像解析室室長・動画解析室室長を務め、画像認識関連のAI特許を在籍中に3件取得。その後、KDDIグループでプロダクトリーダーとして自然言語処理パッケージの自社開発を経て、現在はAGRIST株式会社の執行役員CTO 兼 VPoEとして、農業の人手不足解決に向けた収穫ロボットの開発組織を統括しています。AI・ハード・エレキ・通信・クラウド・IoTまでを一気通貫で設計できる視点を強みに、性能だけでなく「感動やワクワク体験」までデザインできるロボットの研究を進めています。並行して、AI coordinatorとして企業のAI導入・教育機関のAI授業・地域の技術相談を月額契約で継続伴走しています。

AI coordinator

Follow :