dockerでyoloを使える環境を作る

Tweet

dockerイメージをダウンロード

docker pull nvidia/cuda:11.6.0-devel-ubuntu20.04

dockerイメージの実行

docker run --gpus all --name mydarknet -it nvidia/cuda:11.6.0-devel-ubuntu20.04 /bin/bash

dockerで実行したら、以下を実施。

yolov4本家

https://github.com/AlexeyAB/darknet

本家のビルド方法は

git clone https://github.com/AlexeyAB/darknet
cd darknet
mkdir build_release
cd build_release
cmake ..
cmake --build . --target install --parallel 8

とあるが、cmakeがめんどくさかったので

git clone https://github.com/AlexeyAB/darknet
cd darknet
mkdir build_release
cd build_release
make

とやったらビルドはできた。

ビルドしたら、あとはコミットしてプッシュするだけ。
プッシュしたものはkoni0626/darknet2でダウンロードできる。