AlexeyAB/DarknetをNvidia Jetson Nanoにインストール
Raspberry Piとの一番の違いは、GPU対応で、Darknet・AlexeyABをシングルボードコンピューターにインストール出来ることが、大きな魅力となります。
しかも、インストールもWindows版よりも簡単だと思います。
参考記事:Windows10に AlexeyAB・Darknet・YOLO V3導入(Vestal Studio)
ライブラリーアップデート
- apt update
- apt upgrade -y
Cuda関係のパスを環境変数に登録
- export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
- export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
DarknetのダウンロードとYoloモデルのダウンロード
- git clone https://github.com/AlexeyAB/darknet
- cd darknet
- wget https://pjreddie.com/media/files/yolov3.weights
- wget https://pjreddie.com/media/files/yolov3-tiny.weights
MakeFileを変更しGPU、OPENCV対応へ
Makefileを変更し、GPU、CudaとOpencv対応となるようにコンパイルオプションを変更します。.
- sudo vi Makefile
MakeFile変更内容
GPU=1
CUDNN=1
OPENCV=1
Darknetをコンパイル
- make
以上でDarknetが利用可能となります。
darknetフォルダで、以下サンプルコマンドを実行し、動作を確認してください。
- cd darknet
- ./darknet detector test cfg/coco.data yolov3.cfg yolov3.weights -ext_output dog.jpg
Make時にエラーが出て、以下変更を加えました。
変更1
NVCC=nvcc 部分を
NVCC=/usr/local/cuda/bin/nvcc
に変更
変更2 ライブラリ追加
sudo apt install libopencv-dev