Jetson nano에서 YOLOv5 setting

YeongUk·2022년 8월 4일
0

Edge Computing

목록 보기
3/3
post-thumbnail

python설치 및 stats설치

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install python-pip
sudo -H pip install -U jetson-stats
sudo reboot
 
# jetson-stats 실행
jtop

영혼까지 끌어모아서 쓰는 법

메모리가 부족한 jetson시리즈일 경우,
아래와 같이 해주어서 최대한 자원을 아끼고 스왑을 해준다.
(jetson의 메모리가 충분하다면 생략한다.)

1.gdm3 제거 및 lightdm 설치
sudo apt-get install lightdm
sudo apt-get purge gdm3

2.Swap공간 설정하기
sudo apt-get update
sudo apt-get upgrade
 
# nano 에디터를 설치합니다. 이미 설치했다면, 굳이 하지 않아도 좋습니다.
sudo apt-get install nano
 
# dphys-swapfile을 설치합니다.
sudo apt-get install dphys-swapfile
 
3.두 Swap파일의 값이 다음과 같도록 값을 추가하거나, 파일 내 주석을 해제합니다.
# CONF_SWAPSIZE=4096
# CONF_SWAPFACTOR=2
# CONF_MAXSWAP=4096
 
# /sbin/dphys-swapfile를 엽니다.
sudo nano /sbin/dphys-swapfile
 
# 값을 수정한 후 [Ctrl] + [X], [y], [Enter]를 눌러 저장하고 닫습니다
 
 
# /etc/dphys-swapfile를 편집합니다.
sudo nano /etc/dphys-swapfile
 
# 값을 수정한 후 [Ctrl] + [X], [y], [Enter]를 눌러 저장하고 닫습니다
 
4.Jetson Nano 재부팅
sudo reboot

OpenCV, PyTorch, torchvision

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


# PyTorch 1.8.0 다운로드 및 dependencies 설치
wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
 
# Cython, numpy, pytorch 설치
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
 
# torchvision dependencies 설치
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.9.0
python3 setup.py install --user
cd ../ 

YOLOv5

git clone https://github.com/ultralytics/yolov5
cd yolov5
 
# yolov5s.pt weight 다운로드
wget https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt

예제 실행

# 다음 내용 requirements.txt에서 제거
 
numpy>=1.18.5
opencv-python>=4.1.2
torch>=1.7.0
torchvision>=0.8.1
python3 -m pip install --upgrade pip
 
python3 -m pip install -r requirements.txt

pip3 uninstall pillow
pip3 install pillow==8.4.0

python3 detect.py --source 0

profile
소통과 배움을 통해 개인이 가진 맹점을 극복하려 노력하고 있습니다.

3개의 댓글

comment-user-thumbnail
2024년 3월 6일

젯슨나노에서 커스텀학습을 시키기엔 너무 오래 걸릴거 같은데 컴퓨터에서 학습후 bset.pt파일 옮겨 실행하면 학습결과 그대로 반영될까요?

1개의 답글