技術は使ってなんぼ

自分が得たものを誰かの役に立てたい

【メモ】KerasでYOLO v3 を動かすまでの設定手順

【PC環境設定】

1.下記リンクを参考に、Anacondaをインストール
www.python.jp


2.下記リンクを参考に、Anaconda promptを立ち上げ、pipをインストール

gammasoft.jp
→「ライブラリのアップデート」の、> pip install --upgrade pypdf2まで出来ればOK。


3.pipが使えるようになったら、Anaconda promptで、以下のコマンドを入力し、必要なモジュールのインストール。
・Kerasをインストール
 pip install keras
・Tensorflowの1.13.1をインストール
 pip install tensorflow==1.13.1
 pip install tensorflow-gpu==1.13.1


4.CUDA(10.2)とcudnn(7.6.5)を以下URLを参考にインストール
qiita.com


これでPC環境の設定は完了。

【YOLOv3の実行】

以下のURLを参考に、ソースコード一式をダウンロードし、学習済みモデルを取得すれば「python yolo.py」で実行できる
qiita.com


【以下のようなエラーが出た時】

「 E tensorflow/stream_executor/cuda/cuda_dnn.cc:334] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED」

ALLOC_FAILEDであるから、GPUのメモリが足りない。


以下のURLの一番下のページのソースコードを追記する。
satabeblog.com