[Jetson] Jetson Nano๋ฅผ ์ด์šฉํ•œ Yolov5 ํ™œ์šฉ๊ธฐ (3) - yolov5 ์„ค์น˜ ๋ฐ ํ”„๋กœ์ ํŠธ ์ค‘ ๋ฐœ์ƒํ•œ ์˜ค๋ฅ˜(์‹œํ–‰์ฐฉ์˜ค) ์ •๋ฆฌ

โ€ขProjects/Jetson

์ด์ „ ํฌ์ŠคํŒ…์— ์ด์–ด ์ง„ํ–‰ํ•˜๋„๋ก ํ•˜๊ฒ ์Šต๋‹ˆ๋‹ค.
๋“œ๋””์–ด ๋งˆ์ง€๋ง‰ yolov5๋ฅผ ์„ค์น˜ํ•  ์ฐจ๋ก€์ž…๋‹ˆ๋‹ค.

3. Yolov5 ์„ค์น˜

1. github์—์„œ ๊ธฐ๋ณธ repository ๋ฐ›์•„์˜ค๊ธฐ

์ด์ œ 95ํ”„๋กœ ์™”๋‹ค. ๋‹ค์˜จ๊ฑฐ๋‹ค

์•„๋ž˜์˜ ์ฝ”๋“œ๋ฅผ ํ†ตํ•ด Yolov5์˜ repository๋ฅผ ๋ฐ›์•„์ค€๋‹ค.

git clone https://github.com/ultralytics/yolov5
cd yolov5
 
# yolov5s.pt weight ๋‹ค์šด๋กœ๋“œ
wget https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt

2. ์ดํ›„ ํด๋” ๋‚ด์— ์กด์žฌํ•˜๋Š” requirements.txt์—์„œ ์ด๋ฏธ ์„ค์น˜ํ•œ ํ•ญ๋ชฉ๋“ค ์ง€์›Œ์ฃผ๊ธฐ

2-5๊ณผ์ •์—์„œ ์ด๋ฏธ ์„ค์น˜ํ•œ ํ•ญ๋ชฉ๋“ค์ด ์žˆ์œผ๋ฏ€๋กœ requirements.txt์—์„œ ์ œ๊ฑฐํ•ด์ค€๋‹ค.

# ๋‹ค์Œ ๋‚ด์šฉ requirements.txt์—์„œ ์ œ๊ฑฐ
 
numpy>=1.18.5
opencv-python>=4.1.2
torch>=1.7.0
torchvision>=0.8.1

3. pip ์—…๊ทธ๋ ˆ์ด๋“œ ์ดํ›„ requirements.txt ์„ค์น˜

python3 -m pip install --upgrade pip
 
python3 -m pip install -r requirements.txt

์ด๋ ‡๊ฒŒ ํ•˜๋ฉด yolov5 ์„ค์น˜๊ฐ€ ์™„๋ฃŒ๋œ๋‹ค!

4. ์ถ”๊ฐ€ ์ฐธ๊ณ  ์‚ฌํ•ญ

FPS 2๋กœ ์ถœ๋ ฅ๋˜๋Š” ๋ฌธ์ œ (1)

๋‹ค ์„ค์น˜ํ•˜๊ณ  ๋‚˜๋ฉด FPS๊ฐ€ 2๋กœ ์ถœ๋ ฅ๋˜๋ฉด์„œ ์›น์บ ์œผ๋กœ ์˜์ƒ์„ ์ถœ๋ ฅํ•˜๋Š”๋ฐ ๋Š์–ด์ง€๋Š” ํ˜„์ƒ๋“ค์ด ๋ฐœ์ƒํ•œ๋‹ค.

๋ณธ์ธ์˜ ํŒ๋‹จ์œผ๋กœ๋Š” ์•„๋ž˜์˜ ์‚ฌ์ง„์—์„œ ๋‚˜ํƒ€๋‚˜๋Š” ๊ฒฝ๊ณ  ํ•œ ์ค„ ๋•Œ๋ฌธ์ด๋ผ ์ƒ๊ฐ๋˜๋Š”๋ฐ..

Untitled 2

์ผ๋‹จ ๋” ์ฐพ์•„๋ด์•ผํ•  ๋“ฏ ํ•˜๋‹ค.(23.01.12-13)

FPS 2๋กœ ์ถœ๋ ฅ๋˜๋Š” ๋ฌธ์ œ (2) - [inf frames 2560x1472 at 2.00 FPS]

inf frames 2560x1472 at 2.00 FPS ์™€ ๊ฐ™์ด ์ถœ๋ ฅ๋˜๋Š” ๋ถ€๋ถ„์€ utils.dataloader.py์—์„œ ์ˆ˜์ •ํ•ด ์ค„์ˆ˜ ์žˆ๋Š”๋ฐ ์ธ์œ„์ ์œผ๋กœ ์ˆ˜์ •ํ•ด์„œ๋Š” ๊ฐœ์„ ์ด ๋˜์ง€ ์•Š์•˜๋‹ค.

์ € ๋‚ด์šฉ์€ ์•„๋ž˜ ์‚ฌ์ง„๋“ค์—์„œ ๋ณผ ์ˆ˜ ์žˆ๋“ฏ์ด ํ™˜๊ฒฝ์˜ ์ข…๋ฅ˜๊ฐ€ ๋‹ค์–‘ํ•œ๋ฐ, ์ด๋ฅผ ์–ด๋–ป๊ฒŒ ์ ์šฉํ•˜๊ณ  ์ˆ˜์ •ํ•˜๋Š”์ง€ ์ฐพ์•„๋ด์•ผํ•  ๊ฒƒ ๊ฐ™๋‹ค.(23.01.13)

Untitled 3

Untitled 4

์œ„ ๋‘ ๋ฌธ์ œ ํ•ด๊ฒฐ - (23.01.16)

Untitled 5

yolov5/utils์— ์žˆ๋Š” dataloader.py์—์„œ cap = cv2.VideoCapture(s)๋ฅผ cv2.VideoCapture(s, cv2.CAP_V4L)๋กœ ์ž…๋ ฅํ•˜์—ฌ ์ฃผ๋ฉด (1)๋ฒˆ ๋ฌธ์ œ์—์„œ ๋‚˜ํƒ€๋‚˜๋Š” warning ์˜ค๋ฅ˜ ํ•ด๊ฒฐ๋˜๋ฉฐ, ์ด์™€ ๋™์‹œ์— FPS 30์œผ๋กœ ์ฆ๊ฐ€.

๋™์˜์ƒ์˜ ๋Š๊น€์ด ์ „๋ณด๋‹ค๋Š” ๊ฐœ์„ ๋จ์„ ํ™•์ธ.

Untitled 6

(2)๋ฒˆ ๋ฌธ์ œ๋ฅผ ์ฐธ๊ณ ํ•˜์—ฌ inference ํ•ด์ƒ๋„๋ฅผ ์ฆ๊ฐ€์‹œ์ผฐ์„ ๋•Œ 5FPS๋กœ ์ถœ๋ ฅ๋จ.

์ด์— ๋”ฐ๋ผ 640*480์ด ๊ฐ€์žฅ ์šฐ์ˆ˜ํ•˜๋‹ค๊ณ  ํŒ๋‹จ๋จ.

๊ธ€์„ ๋งˆ๋ฌด๋ฆฌํ•˜๋ฉฐ,,

์ด๋ ‡๊ฒŒ ํ”„๋กœ์ ํŠธ๊ฐ€ ๋งˆ๋ฌด๋ฆฌ ๋ฉ๋‹ˆ๋‹ค.
๋ณธ ํ”„๋กœ์ ํŠธ์—์„œ dataset์„ ์ œ์ž‘ํ•  ๋•Œ์—๋Š” roboflow ์‚ฌ์ดํŠธ ๋‚ด์—์„œ ๋ฐ์ดํ„ฐ๋ฅผ ์ง์ ‘ ์ œ์ž‘ํ•˜์—ฌ trainํ•˜๋Š”๋ฐ ์‚ฌ์šฉ์„ ํ–ˆ์Šต๋‹ˆ๋‹ค.

๊ธฐํšŒ๊ฐ€ ๋œ๋‹ค๋ฉด ๋ฐ์ดํ„ฐ ์ œ์ž‘๊ธฐ์— ๋Œ€ํ•ด์„œ๋„ ์ •๋ฆฌ๋ฅผ ํ•ด์•ผ๊ฒ ๋„ค์š”.

์•„๋ฌดํŠผ 2022๋…„์˜ 6๊ฐœ์›” ์ด์ƒ ์†Œ์š”๋˜์—ˆ๋˜ yolov5 with Jetson Nano ํ”„๋กœ์ ํŠธ๋Š” ๋‹ค์Œ๊ณผ ๊ฐ™์ด ๋งˆ๋ฌด๋ฆฌ ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.
์‚ฌ์šฉํ•œ ์ œํ’ˆ์ด 4GB delveloper kit์ด๋‹ค๋ณด๋‹ˆ, ์„ฑ๋Šฅ์  ์ธก๋ฉด์—์„œ ํ›Œ๋ฅญํ•œ ๊ธฐ๋Œ€ํšจ๊ณผ๋ฅผ ๋ณผ ์ˆ˜๋Š” ์—†์ง€๋งŒ, ๋ฐ์ดํ„ฐ์˜ ํŠน์ง•์ด ๋šœ๋ ทํ•˜๋‹ค๋ฉด ์ถฉ๋ถ„ํžˆ ์‚ฌ์šฉ๊ฐ€๋Šฅํ•  ๊ฒƒ์œผ๋กœ ํŒ๋‹จ๋ฉ๋‹ˆ๋‹ค.

๋˜ํ•œ yahboom์˜ ๊ฒฝ์šฐ yolov4๋ฅผ ๊ธฐ๋ณธ์œผ๋กœ ์ง€์›ํ•˜๋Š”๋ฐ, ์ฒ˜์Œ ํ”„๋กœ์ ํŠธ๋ฅผ nvidia ์ œํ’ˆ์œผ๋กœ ์ง„ํ–‰ํ•˜์˜€๊ณ , ๊ทธ ๊ณผ์ •์—์„œ yolov5๋ฅผ ์‚ฌ์šฉํ•˜์˜€๊ธฐ ๋•Œ๋ฌธ์— ํ•˜๋Š” ์ˆ˜ ์—†์ด ์ด๋Ÿฐ ์—ฌ๋Ÿฌ ์ˆ˜๊ณ ๋“ค์ด ์žˆ์—ˆ์Šต๋‹ˆ๋‹ค.
์ž๋ฃŒ ๋˜ํ•œ ํ˜„์ €ํžˆ ์ ์—ˆ๊ตฌ์š”...
๋ˆ„๊ตฐ๊ฐ€ ๋ณธ ํฌ์ŠคํŒ…๋“ค์„ ํ†ตํ•ด ํ•ด๊ฒฐ์ด ๋œ๋‹ค๋ฉด ๊ทธ๊ฑธ๋กœ ์ •๋ฆฌํ•˜๋Š” ์ž…์žฅ์—์„œ๋Š” ๋งŒ์กฑ์Šค๋Ÿฌ์šธ ๊ฒƒ ๊ฐ™์Šต๋‹ˆ๋‹ค!

PS. ์ถ”๊ฐ€ ๋ฌธ์˜์‚ฌํ•ญ ๋ฐ ์งˆ๋ฌธ์€ ํ™˜์˜ํ•ฉ๋‹ˆ๋‹ค. ๊ทธ๋ฅผ ํ†ตํ•ด ์ €๋„ ๋” ์„ฑ์žฅํ•  ์ˆ˜ ์žˆ์„ํ…Œ๋‹ˆ๊นŒ์š”. ๊ธด ๊ธ€ ์ฝ์–ด์ฃผ์…”์„œ ๊ฐ์‚ฌํ•ฉ๋‹ˆ๋‹ค.

Share