YOLOv5 on Jetson Nano

2022.09.28

この記事ではYOLOv5をJetson Nanoで動作させる方法について解説します。YOLOv5はいくつかの実装がありますがここではオリジナルのPyTorch実装YOLOv5を動作させる方法について解説します。

YOLOv5はpython3.7以上を必要としますが、Jetson Nanoはpython3.6がデフォルトでインストールされていて、公式のPyTorchライブラリもpython3.6の為これらは使用できません。

ですのでこちらの記事を参考にpython3.8で使用できるPyTorchライブラリを導入します。

実は難しいのは上記のカスタムPyTorchの導入で、YOLOv5の導入自体は特に難しくありません。

git clone https://github.com/ultralytics/yolov5.git
cd yolov5
python3.8 -m pip install -r requirements.txt
python3.8 detect.py --source 0

--sourceでvideo sourceを指定しています。他にも様々なオプションがありますので

python3.8 detect.py --help

として調べてみてください。

以上です。