画像認識で遊ぶ - 地平線まで行ってくる。

Yoloを使うと割とお手軽に画像認識が体験できると聞いて、チャレンジ。

まずは、

次に

次に、Anacondaのコマンドラインを使ってPyTorchをインストール。環境に応じてコマンドオプションを指示してくれる。素晴らしい。

と、その前に作業用フォルダを作って、そこで作業を続けます・・・。

　OpenCV忘れてたので・・・

　conda install opencv

　でOpenCVもインストール。

　続いて pytorch用のyolo v3を有り難く収得。

　git clone https://github.com/ayooshkathuria/pytorch-yolo-v3.git

　学習済みのデータ・セットも有り難くいただく。YOLO本家のサイトにあります。　pjreddie.com

そのうち、

を選んでダウンロードした後に作業フォルダに同じく入れ込む。

　USBカメラを引っ張り出してきて

　python cam_demo.py

　ですんなり起動・・・。できた！

　GPUない貧弱PCなので、CUDAはなしです。が、こんなに簡単にお試しできたのでびっくり。

　次は、教示データ作り！？

参考：