Jetson Nanoと Jetson Xavier NXで YOLOv5を動かす

最新版をご覧ください。

• Jetson Nano 4GB
• Jetson Xavier NX
• logicool C270N
• JetPack 4.6

参考：https://qiita.com/ryryry/items/d95c2207b9608243340e

パッケージの更新

sudo apt update
sudo apt upgrade
（Configuring docker.io -> Yes）

OpenCVのインストール

「apt install python3-opencv」ではweb camをソースとした時にSegmentation faultエラーとなったため、ソースコードからインストールしました。

スワップ領域の設定（Xavier NXでは不要です）

sudo apt install dphys-swapfile
sudo nano /etc/dphys-swapfile

/etc/dphys-swapfile

CONF_SWAPSIZE=4096
CONF_MAXSWAP=4096

reboot

OpenCVのインストール

wget https://github.com/Qengineering/Install-OpenCV-Jetson-Nano/raw/main/OpenCV-4-5-0.sh
sudo chmod 755 ./OpenCV-4-5-0.sh
./OpenCV-4-5-0.sh
rm OpenCV-4-5-0.sh

参考：https://qengineering.eu/install-opencv-4.5-on-jetson-nano.html

こちらもご参照ください

torchのインストール

• torch（PyTorch）は、Facebookが開発を主導したPython向けの機械学習ライブラリです。

wget https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl -O torch-1.7.0-cp36-cp36m-linux_aarch64.whl
sudo apt install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
pip3 install numpy torch-1.7.0-cp36-cp36m-linux_aarch64.whl

torchvisionのインストール

• tourchvisionは、画像処理におけるデータセット（MNIST、COCO）、モデルアーキテクチャ、画像変換処理から構成されたパッケージです。

git clone --branch v0.8.1 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.x.0
sudo python3 setup.py install
cd

YOLOv5のインストール

pip3 install -U pip
pip3 install -U setuptools

git clone https://github.com/ultralytics/yolov5.git

インストール済みのパッケージは事前に無効にしておきます。

requirements.txt

#opencv-python>=4.1.2
#torch>=1.7.0
#tensorboard>=2.2
#torchvision>=0.8.1

pip3 install -r requirements.txt

YOLOv5の動作確認

• detect.pyの実行

python3 detect.py

初回起動時はウェイトファイル「yolov5s.pt」がダウンロードされます。

検出結果は「yolov5/runs/detect/exp」内に保存されます（bus.jpg、zidane.jpg）。

• webカメラから動画を入力して認識を実行（yolov5s.ptを使用)

python3 detect.py --source 0 --imgsz=256 --nosave

Jetson Nano：1 frame：0.063s（15.8fps）

Jetson Xavier NX（MODE 20W 6CORE）：1 frame：0.029-0.033s（30.3fps）

この後は