์ผ๋ฐ์ ์ธ YOLO ๋ฌธ์ ํด๊ฒฐ
์๊ฐ
์ด ๊ฐ์ด๋๋ Ultralytics ํ๋ก์ ํธ์์ YOLO11 ๋ก ์์ ํ๋ ๋์ ๋ฐ์ํ๋ ์ผ๋ฐ์ ์ธ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋ ๋ฐ ๋์์ด ๋๋ ์ข ํฉ์ ์ธ ์๋ด์์ ๋๋ค. ์ฌ๋ฐ๋ฅธ ์๋ด๋ฅผ ํตํด ์ด๋ฌํ ๋ฌธ์ ๋ฅผ ์ฝ๊ฒ ํด๊ฒฐํ ์ ์์ผ๋ฏ๋ก ๋ถํ์ํ ์ง์ฐ ์์ด ํ๋ก์ ํธ๋ฅผ ์ ์์ ์ผ๋ก ์งํํ ์ ์์ต๋๋ค.
Watch: Ultralytics YOLO11 ์ผ๋ฐ์ ์ธ ๋ฌธ์ | ์ค์น ์ค๋ฅ, ๋ชจ๋ธ ๊ต์ก ๋ฌธ์
์ผ๋ฐ์ ์ธ ๋ฌธ์
์ค์น ์ค๋ฅ
์ค์น ์ค๋ฅ๋ ํธํ๋์ง ์๋ ๋ฒ์ , ์ข ์์ฑ ๋๋ฝ, ์๋ชป๋ ํ๊ฒฝ ์ค์ ๋ฑ ๋ค์ํ ์์ธ์ผ๋ก ์ธํด ๋ฐ์ํ ์ ์์ต๋๋ค. ๋จผ์ ๋ค์์ ์ํํ๊ณ ์๋์ง ํ์ธํ์ธ์:
-
๊ถ์ฅ๋๋ Python 3.8 ์ด์์ ์ฌ์ฉ ์ค์ ๋๋ค.
-
์ฌ๋ฐ๋ฅธ ๋ฒ์ ์ PyTorch (1.8 ์ด์)์ด ์ค์น๋์ด ์๋์ง ํ์ธํฉ๋๋ค.
-
์ถฉ๋์ ํผํ๊ธฐ ์ํด ๊ฐ์ ํ๊ฒฝ์ ์ฌ์ฉํ๋ ๊ฒ์ ๊ณ ๋ คํ์ธ์.
-
๊ณต์ ์ค์น ๊ฐ์ด๋๋ฅผ ๋จ๊ณ๋ณ๋ก ๋ฐ๋ผํ์ธ์.
๋ํ ์ฌ์ฉ์๋ค์ด ํํ ๊ฒช๋ ์ค์น ๋ฌธ์ ์ ๊ฐ๊ฐ์ ํด๊ฒฐ ๋ฐฉ๋ฒ์ ๋ค์๊ณผ ๊ฐ์ต๋๋ค:
-
๊ฐ์ ธ์ค๊ธฐ ์ค๋ฅ ๋๋ ์ข ์์ฑ ๋ฌธ์ - YOLO11 ๋ฅผ ๊ฐ์ ธ์ค๋ ๋์ ์ค๋ฅ๊ฐ ๋ฐ์ํ๊ฑฐ๋ ์ข ์์ฑ๊ณผ ๊ด๋ จ๋ ๋ฌธ์ ๊ฐ ์๋ ๊ฒฝ์ฐ ๋ค์ ๋ฌธ์ ํด๊ฒฐ ๋จ๊ณ๋ฅผ ๊ณ ๋ คํ์ธ์:
-
์๋ก ์ค์น: ์๋ก ์ค์นํ๋ฉด ์๊ธฐ์น ์์ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ ์ ์๋ ๊ฒฝ์ฐ๊ฐ ์์ต๋๋ค. ํนํ Ultralytics ์ ๊ฐ์ ๋ผ์ด๋ธ๋ฌ๋ฆฌ์ ๊ฒฝ์ฐ ์ ๋ฐ์ดํธ๋ก ์ธํด ํ์ผ ํธ๋ฆฌ ๊ตฌ์กฐ๋ ๊ธฐ๋ฅ์ด ๋ณ๊ฒฝ๋ ์ ์์ต๋๋ค.
-
์ ๊ธฐ์ ์ผ๋ก ์ ๋ฐ์ดํธํฉ๋๋ค: ์ต์ ๋ฒ์ ์ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ฌ์ฉํ๊ณ ์๋์ง ํ์ธํ์ธ์. ์ด์ ๋ฒ์ ์ ์ต์ ์ ๋ฐ์ดํธ์ ํธํ๋์ง ์์ ์ ์ฌ์ ์ธ ์ถฉ๋์ด๋ ๋ฌธ์ ๊ฐ ๋ฐ์ํ ์ ์์ต๋๋ค.
-
์ข ์์ฑ์ ํ์ธํฉ๋๋ค: ๋ชจ๋ ํ์ ์ข ์์ฑ์ด ์ฌ๋ฐ๋ฅด๊ฒ ์ค์น๋์ด ์๊ณ ํธํ๋๋ ๋ฒ์ ์ธ์ง ํ์ธํฉ๋๋ค.
-
๋ณ๊ฒฝ ์ฌํญ ๊ฒํ : ์ด์ ๋ฒ์ ์ ์ฒ์ ๋ณต์ ํ๊ฑฐ๋ ์ค์นํ ๊ฒฝ์ฐ, ์ค์ํ ์ ๋ฐ์ดํธ๊ฐ ๋ผ์ด๋ธ๋ฌ๋ฆฌ์ ๊ตฌ์กฐ๋ ๊ธฐ๋ฅ์ ์ํฅ์ ๋ฏธ์น ์ ์๋ค๋ ์ ์ ์ ์ํ์ธ์. ํญ์ ๊ณต์ ๋ฌธ์๋ ๋ณ๊ฒฝ ๋ก๊ทธ๋ฅผ ์ฐธ์กฐํ์ฌ ์ฃผ์ ๋ณ๊ฒฝ ์ฌํญ์ ํ์ ํ์ธ์.
-
์ํํ๊ณ ์ค๋ฅ ์๋ ๊ฒฝํ์ ์ํด์๋ ๋ผ์ด๋ธ๋ฌ๋ฆฌ์ ์ข ์์ฑ์ ์ต์ ์ํ๋ก ์ ์งํ๋ ๊ฒ์ด ์ค์ํฉ๋๋ค.
-
-
GPU ์์ YOLO11 ์คํ - YOLO11 ๋ฅผ GPU ์์ ์คํํ๋ ๋ฐ ๋ฌธ์ ๊ฐ ์๋ ๊ฒฝ์ฐ ๋ค์ ๋ฌธ์ ํด๊ฒฐ ๋จ๊ณ๋ฅผ ๊ณ ๋ คํ์ธ์:
-
CUDA ํธํ์ฑ ๋ฐ ์ค์น ํ์ธ: GPU ์ด CUDA ํธํ๋๋์ง, CUDA ์ด ์ฌ๋ฐ๋ฅด๊ฒ ์ค์น๋์๋์ง ํ์ธํฉ๋๋ค. ์ฌ์ฉ
nvidia-smi
๋ช ๋ น์ ์ฌ์ฉํ์ฌ NVIDIA GPU ๋ฐ CUDA ๋ฒ์ ์ ์ํ๋ฅผ ํ์ธํ ์ ์์ต๋๋ค. -
PyTorch ๋ฐ CUDA ํตํฉ ํ์ธ: PyTorch ์ ์คํํ์ฌ CUDA ์ ํ์ฉํ ์ ์๋์ง ํ์ธ
import torch; print(torch.cuda.is_available())
๋ฅผ Python ํฐ๋ฏธ๋์์ ์คํํฉ๋๋ค. 'True'๋ฅผ ๋ฐํํ๋ฉด PyTorch ์ CUDA ์ ์ฌ์ฉํ๋๋ก ์ค์ ๋ฉ๋๋ค. -
ํ๊ฒฝ ํ์ฑํ: ํ์ํ ๋ชจ๋ ํจํค์ง๊ฐ ์ค์น๋์ด ์๋ ์ฌ๋ฐ๋ฅธ ํ๊ฒฝ์ธ์ง ํ์ธํฉ๋๋ค.
-
ํจํค์ง ์ ๋ฐ์ดํธ: ์ค๋๋ ํจํค์ง๋ GPU ๊ณผ ํธํ๋์ง ์์ ์ ์์ต๋๋ค. ๊ณ์ ์ ๋ฐ์ดํธํ์ธ์.
-
ํ๋ก๊ทธ๋จ ๊ตฌ์ฑ: ํ๋ก๊ทธ๋จ ๋๋ ์ฝ๋์ GPU ์ฌ์ฉ๋ฒ์ด ๋ช ์๋์ด ์๋์ง ํ์ธํฉ๋๋ค. YOLO11 ์์ ์ค์ ๋๋ ๊ตฌ์ฑ์ ์์ ์ ์์ต๋๋ค.
-
๋ชจ๋ธ ๊ต์ก ๋ฌธ์
์ด ์น์ ์์๋ ๊ต์ก ์ค์ ์ง๋ฉดํ๋ ์ผ๋ฐ์ ์ธ ๋ฌธ์ ์ ๊ทธ์ ๋ํ ์ค๋ช ๋ฐ ํด๊ฒฐ ๋ฐฉ๋ฒ์ ๋ค๋ฃน๋๋ค.
๊ตฌ์ฑ ์ค์ ํ์ธ
์ด์: ๊ตฌ์ฑ ์ค์ ์ด ์ฌ๋ฐ๋ฅธ์ง ํ์คํ์ง ์์ต๋๋ค. .yaml
ํ์ผ์ด ๋ชจ๋ธ ํ์ต ์ค์ ์ฌ๋ฐ๋ฅด๊ฒ ์ ์ฉ๋๊ณ ์๋์ง ํ์ธํฉ๋๋ค.
์๋ฃจ์
: ๊ตฌ์ฑ ์ค์ ์ .yaml
ํ์ผ์ ์ฌ์ฉํ ๋ ์ ์ฉํด์ผ ํฉ๋๋ค. model.train()
๊ธฐ๋ฅ์ ์ฌ์ฉํ์ธ์. ์ด๋ฌํ ์ค์ ์ด ์ฌ๋ฐ๋ฅด๊ฒ ์ ์ฉ๋์๋์ง ํ์ธํ๋ ค๋ฉด ๋ค์ ๋จ๊ณ๋ฅผ ๋ฐ๋ฅด์ธ์:
- ๊ฒฝ๋ก๊ฐ
.yaml
๊ตฌ์ฑ ํ์ผ์ด ์ฌ๋ฐ๋ฅธ์ง ํ์ธํฉ๋๋ค. - ๊ฒฝ๋ก๋ฅผ ์ ๋ฌํด์ผ ํฉ๋๋ค.
.yaml
ํ์ผ์data
์ธ์๋ฅผ ํธ์ถํ ๋model.train()
๋ฅผ ํด๋ฆญํฉ๋๋ค:
๋ค์ค GPU๋ก ํ๋ จ ๊ฐ์ํ
๋ฌธ์ : ๋จ์ผ GPU ์์ ํธ๋ ์ด๋ ์๋๊ฐ ๋๋ฆฌ๊ณ ์ฌ๋ฌ GPU๋ฅผ ์ฌ์ฉํ์ฌ ํ๋ก์ธ์ค ์๋๋ฅผ ๋์ด๊ณ ์ถ์ต๋๋ค.
์๋ฃจ์ : ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ๋๋ฆฌ๋ฉด ํ์ต ์๋๋ฅผ ๋์ผ ์ ์์ง๋ง GPU ๋ฉ๋ชจ๋ฆฌ ์ฉ๋์ ๊ณ ๋ คํด์ผ ํฉ๋๋ค. ์ฌ๋ฌ ๊ฐ์ GPU๋ก ํ๋ จ ์๋๋ฅผ ๋์ด๋ ค๋ฉด ๋ค์ ๋จ๊ณ๋ฅผ ๋ฐ๋ฅด์ธ์:
-
์ฌ๋ฌ ๊ฐ์ GPU๋ฅผ ์ฌ์ฉํ ์ ์๋์ง ํ์ธํฉ๋๋ค.
-
.yaml ๊ตฌ์ฑ ํ์ผ์ ์์ ํ์ฌ ์ฌ์ฉํ GPU ์(์: gpus)๋ฅผ ์ง์ ํฉ๋๋ค: 4.
-
๋ฉ๋ชจ๋ฆฌ ์ ํ์ ์ด๊ณผํ์ง ์๊ณ ์ฌ๋ฌ GPU๋ฅผ ์ต๋ํ ํ์ฉํ๋ ค๋ฉด ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ์ ์ ํ ๋๋ฆฌ์ธ์.
-
์ฌ๋ฌ GPU๋ฅผ ํ์ฉํ๋๋ก ํ๋ จ ๋ช ๋ น์ ์์ ํฉ๋๋ค:
# Adjust the batch size and other settings as needed to optimize training speed
model.train(data="/path/to/your/data.yaml", batch=32, multi_scale=True)
์ฐ์ ๋ชจ๋ํฐ๋ง ๋งค๊ฐ๋ณ์
๋ฌธ์ : ์์ค๊ณผ๋ ๋ณ๋๋ก ํ๋ จ ์ค์ ์ง์์ ์ผ๋ก ๋ชจ๋ํฐ๋งํด์ผ ํ๋ ๋งค๊ฐ๋ณ์๋ฅผ ์๊ณ ์ถ์ต๋๋ค.
์๋ฃจ์ : ์์ค์ ๋ชจ๋ํฐ๋งํด์ผ ํ ์ค์ํ ์งํ์ด์ง๋ง, ๋ชจ๋ธ ์ฑ๋ฅ ์ต์ ํ๋ฅผ ์ํด ๋ค๋ฅธ ์งํ๋ ์ถ์ ํ๋ ๊ฒ์ด ํ์์ ์ ๋๋ค. ํ๋ จ ์ค์ ๋ชจ๋ํฐ๋งํด์ผ ํ ๋ช ๊ฐ์ง ์ฃผ์ ์งํ๋ ๋ค์๊ณผ ๊ฐ์ต๋๋ค:
- ์ ๋ฐ๋
- ๋ฆฌ์ฝ
- ํ๊ท ํ๊ท ์ ๋ฐ๋ (mAP)
์ด๋ฌํ ๋ฉํธ๋ฆญ์ ํธ๋ ์ด๋ ๋ก๊ทธ์์ ๋๋ ์๊ฐํ๋ฅผ ์ํด TensorBoard ๋๋ wandb์ ๊ฐ์ ๋๊ตฌ๋ฅผ ์ฌ์ฉํ์ฌ ์ก์ธ์คํ ์ ์์ต๋๋ค. ์ด๋ฌํ ๋ฉํธ๋ฆญ์ ๊ธฐ๋ฐ์ผ๋ก ์กฐ๊ธฐ ์ค์ง๋ฅผ ๊ตฌํํ๋ฉด ๋ ๋์ ๊ฒฐ๊ณผ๋ฅผ ์ป์ ์ ์์ต๋๋ค.
๊ต์ก ์งํ ์ํฉ์ ์ถ์ ํ๋ ๋๊ตฌ
์ด์: ๊ต์ก ์งํ ์ํฉ์ ์ถ์ ํ ์ ์๋ ๋๊ตฌ์ ๋ํ ์ถ์ฒ์ ์ฐพ๊ณ ์์ต๋๋ค.
์๋ฃจ์ : ๊ต์ก ์งํ ์ํฉ์ ์ถ์ ํ๊ณ ์๊ฐํํ๋ ค๋ฉด ๋ค์ ๋๊ตฌ๋ฅผ ์ฌ์ฉํ๋ ๊ฒ์ ๊ณ ๋ คํด ๋ณผ ์ ์์ต๋๋ค:
- ํ ์๋ณด๋: TensorBoard๋ ์์ค, ์ ํ๋ ๋ฑ์ ํฌํจํ ๊ต์ก ์งํ๋ฅผ ์๊ฐํํ๋ ๋ฐ ๋๋ฆฌ ์ฌ์ฉ๋๋ ๋๊ตฌ์ ๋๋ค. YOLO11 ๊ต์ก ํ๋ก์ธ์ค์ ํตํฉํ ์ ์์ต๋๋ค.
- CometComet ์์๋ ์คํ ์ถ์ ๋ฐ ๋น๊ต๋ฅผ ์ํ ๊ด๋ฒ์ํ ํดํท์ ์ ๊ณตํฉ๋๋ค. ์ด๋ฅผ ํตํด ๋ฉํธ๋ฆญ, ํ์ดํผํ๋ผ๋ฏธํฐ, ๋ชจ๋ธ ๊ฐ์ค์น๊น์ง ์ถ์ ํ ์ ์์ต๋๋ค. YOLO ๋ชจ๋ธ๊ณผ์ ํตํฉ๋ ๊ฐ๋จํ์ฌ ์คํ ์ฃผ๊ธฐ์ ๋ํ ์ ์ฒด ๊ฐ์๋ฅผ ์ ๊ณตํฉ๋๋ค.
- Ultralytics ํ๋ธ: Ultralytics ํ๋ธ๋ YOLO ๋ชจ๋ธ ์ถ์ ์ ํนํ๋ ํ๊ฒฝ์ ์ ๊ณตํ์ฌ ๋ฉํธ๋ฆญ, ๋ฐ์ดํฐ ์ธํธ๋ฅผ ๊ด๋ฆฌํ๊ณ ํ๊ณผ ํ์ ํ ์ ์๋ ์์คํฑ ํ๋ซํผ์ ์ ๊ณตํฉ๋๋ค. YOLO ์ ์ด์ ์ ๋ง์ถ ๋ง์ถคํ ์ถ์ ์ต์ ์ ์ ๊ณตํฉ๋๋ค.
์ด๋ฌํ ๋๊ตฌ๋ ๊ฐ๊ฐ ๊ณ ์ ํ ์ฅ์ ์ ์ ๊ณตํ๋ฏ๋ก ์ ํํ ๋ ํ๋ก์ ํธ์ ํน์ ์๊ตฌ ์ฌํญ์ ๊ณ ๋ คํ ์ ์์ต๋๋ค.
๊ต์ก์ด ์งํ ์ค์ธ์ง ํ์ธํ๋ ๋ฐฉ๋ฒ GPU
๋ฌธ์ ์ ๋๋ค: ํธ๋ ์ด๋ ๋ก๊ทธ์ '์ฅ์น' ๊ฐ์ด 'null'์ด๋ฉฐ, GPU ์์ ํธ๋ ์ด๋์ด ์งํ๋๊ณ ์๋์ง ํ์คํ์ง ์์ต๋๋ค.
ํด๊ฒฐ ๋ฐฉ๋ฒ: 'device' ๊ฐ์ด 'null'์ด๋ผ๋ ๊ฒ์ ์ผ๋ฐ์ ์ผ๋ก ํธ๋ ์ด๋ ํ๋ก์ธ์ค๊ฐ ๊ธฐ๋ณธ ๋์์ธ ์ฌ์ฉ ๊ฐ๋ฅํ GPU ์ ์๋์ผ๋ก ์ฌ์ฉํ๋๋ก ์ค์ ๋์ด ์์์ ์๋ฏธํฉ๋๋ค. ํน์ GPU ์์ ๊ต์ก์ด ์ด๋ฃจ์ด์ง๋๋ก ํ๋ ค๋ฉด .yaml ๊ตฌ์ฑ ํ์ผ์์ 'device' ๊ฐ์ GPU ์ธ๋ฑ์ค(์: ์ฒซ ๋ฒ์งธ GPU ์ ๊ฒฝ์ฐ '0')๋ก ์๋์ผ๋ก ์ค์ ํ ์ ์์ต๋๋ค:
์ด๋ ๊ฒ ํ๋ฉด ์ง์ ๋ GPU ์ ๊ต์ก ํ๋ก์ธ์ค๊ฐ ๋ช ์์ ์ผ๋ก ํ ๋น๋ฉ๋๋ค. CPU ์์ ๊ต์กํ๋ ค๋ฉด 'device'๋ฅผ 'cpu'๋ก ์ค์ ํฉ๋๋ค.
'์คํ' ํด๋์์ ๋ก๊ทธ์ ๋ฉํธ๋ฆญ์ ์ฃผ์ํ์ฌ ๊ต์ก ์งํ ์ํฉ์ ํจ๊ณผ์ ์ผ๋ก ๋ชจ๋ํฐ๋งํ์ธ์.
ํจ๊ณผ์ ์ธ ๋ชจ๋ธ ํธ๋ ์ด๋์ ์ํ ์ฃผ์ ๊ณ ๋ ค ์ฌํญ
๋ค์์ ๋ชจ๋ธ ํธ๋ ์ด๋๊ณผ ๊ด๋ จ๋ ๋ฌธ์ ์ ์ง๋ฉดํ์ ๋ ์ผ๋์ ๋์ด์ผ ํ ๋ช ๊ฐ์ง ์ฌํญ์ ๋๋ค.
๋ฐ์ดํฐ ์ธํธ ํ์ ๋ฐ ๋ ์ด๋ธ
-
์ค์์ฑ: ๋ชจ๋ ๋จธ์ ๋ฌ๋ ๋ชจ๋ธ์ ๊ธฐ์ด๋ ํ์ต๋๋ ๋ฐ์ดํฐ์ ํ์ง๊ณผ ํ์์ ์์ต๋๋ค.
-
๊ถ์ฅ ์ฌํญ: ์ฌ์ฉ์ ์ง์ ๋ฐ์ดํฐ ์งํฉ๊ณผ ๊ด๋ จ ๋ ์ด๋ธ์ด ์์ ํ์์ ์ค์ํ๋์ง ํ์ธํ์ธ์. ์ฃผ์์ด ์ ํํ๊ณ ๊ณ ํ์ง์ธ์ง ํ์ธํ๋ ๊ฒ์ด ์ค์ํฉ๋๋ค. ์ฃผ์์ด ๋ถ์ ํํ๊ฑฐ๋ ์์ค ์ดํ์ ์ฃผ์์ ๋ชจ๋ธ์ ํ์ต ํ๋ก์ธ์ค๋ฅผ ๋ฐฉํดํ์ฌ ์์ธกํ ์ ์๋ ๊ฒฐ๊ณผ๋ฅผ ์ด๋ํ ์ ์์ต๋๋ค.
๋ชจ๋ธ ์ปจ๋ฒ์ ์ค
-
์ค์์ฑ: ๋ชจ๋ธ ์๋ ด์ ๋ฌ์ฑํ๋ฉด ๋ชจ๋ธ์ด ํ์ต ๋ฐ์ดํฐ๋ก๋ถํฐ ์ถฉ๋ถํ ํ์ตํ๋์ง ํ์ธํ ์ ์์ต๋๋ค.
-
๊ถ์ฅ ์ฌํญ: ๋ชจ๋ธ์ '์ฒ์๋ถํฐ' ํ๋ จํ ๋๋ ๋ชจ๋ธ์ด ๋ง์กฑํ ๋งํ ์์ค์ ์๋ ด์ ๋๋ฌํ๋๋ก ํ๋ ๊ฒ์ด ์ค์ํฉ๋๋ค. ์ด๋ฅผ ์ํด์๋ ๊ธฐ์กด ๋ชจ๋ธ์ ๋ฏธ์ธ ์กฐ์ ํ ๋๋ณด๋ค ๋ ๊ธด ํ๋ จ ๊ธฐ๊ฐ๊ณผ ๋ ๋ง์ ์ํฌํฌ๊ฐ ํ์ํ ์ ์์ต๋๋ค.
ํ์ต ์๋ ๋ฐ ๋ฐฐ์น ํฌ๊ธฐ
-
์ค์์ฑ: ์ด๋ฌํ ํ์ดํผํ๋ผ๋ฏธํฐ๋ ํ์ต ์ค์ ๋ชจ๋ธ์ด ๊ฐ์ค์น๋ฅผ ์ ๋ฐ์ดํธํ๋ ๋ฐฉ๋ฒ์ ๊ฒฐ์ ํ๋ ๋ฐ ์ค์ถ์ ์ธ ์ญํ ์ ํฉ๋๋ค.
-
๊ถ์ฅ ์ฌํญ: ์ ํํ ํ์ต ์๋์ ๋ฐฐ์น ํฌ๊ธฐ๊ฐ ํน์ ๋ฐ์ดํฐ ์ธํธ์ ์ต์ ์ธ์ง ์ ๊ธฐ์ ์ผ๋ก ํ๊ฐํ์ธ์. ๋ฐ์ดํฐ ์ธํธ์ ํน์ฑ๊ณผ ์ผ์นํ์ง ์๋ ๋งค๊ฐ๋ณ์๋ ๋ชจ๋ธ์ ์ฑ๋ฅ์ ์ ํดํ ์ ์์ต๋๋ค.
ํด๋์ค ๋ฐฐํฌ
-
์ค์์ฑ: ๋ฐ์ดํฐ ์งํฉ์ ํด๋์ค ๋ถํฌ๋ ๋ชจ๋ธ์ ์์ธก ๊ฒฝํฅ์ ์ํฅ์ ๋ฏธ์น ์ ์์ต๋๋ค.
-
๊ถ์ฅ ์ฌํญ: ๋ฐ์ดํฐ ์งํฉ ๋ด์ ํด๋์ค ๋ถํฌ๋ฅผ ์ ๊ธฐ์ ์ผ๋ก ํ๊ฐํ์ธ์. ํด๋์ค ๋ถ๊ท ํ์ด ์๋ ๊ฒฝ์ฐ, ๋ชจ๋ธ์ด ๋ ๋ง์ด ๋ถํฌํ๋ ํด๋์ค์ ํธํฅ๋ ์ํ์ด ์์ต๋๋ค. ์ด๋ฌํ ํธํฅ์ ํผ๋ ํ๋ ฌ์์ ๋ถ๋ช ํ๊ฒ ๋๋ฌ๋ ์ ์์ผ๋ฉฐ, ์ฌ๊ธฐ์ ๋ชจ๋ธ์ ์ฃผ๋ก ๋ค์ ํด๋์ค๋ฅผ ์์ธกํ ์ ์์ต๋๋ค.
์ฌ์ ํ์ต๋ ๊ฐ์ค์น๋ก ๊ต์ฐจ ํ์ธ
-
์ค์์ฑ: ์ฌ์ ํ์ต๋ ๊ฐ์ค์น๋ฅผ ํ์ฉํ๋ฉด ํนํ ๋ฐ์ดํฐ๊ฐ ์ ํ์ ์ธ ๊ฒฝ์ฐ ๋ชจ๋ธ ํ์ต์ ์ํ ํ์คํ ์ถ๋ฐ์ ์ ์ ๊ณตํ ์ ์์ต๋๋ค.
-
๊ถ์ฅ ์ฌํญ: ์ง๋จ ๋จ๊ณ๋ก ๋์ผํ ๋ฐ์ดํฐ๋ฅผ ์ฌ์ฉํ์ฌ ๋ชจ๋ธ์ ํ๋ จํ๋ ๋ฏธ๋ฆฌ ํ๋ จ๋ ๊ฐ์ค์น๋ก ์ด๊ธฐํํ๋ ๊ฒ์ ๊ณ ๋ คํด ๋ณด์ธ์. ์ด ์ ๊ทผ ๋ฐฉ์์ ํตํด ํผ๋ ํ๋ ฌ์ด ์ ํ์ฑ๋๋ฉด '์ฒ์๋ถํฐ' ๋ชจ๋ธ์ ์ถ๊ฐ ํ์ต์ด๋ ์กฐ์ ์ด ํ์ํ ์ ์์์ ์์ฌํ ์ ์์ต๋๋ค.
๋ชจ๋ธ ์์ธก๊ณผ ๊ด๋ จ๋ ๋ฌธ์
์ด ์น์ ์์๋ ๋ชจ๋ธ ์์ธก ์ค์ ์ง๋ฉดํ๋ ์ผ๋ฐ์ ์ธ ๋ฌธ์ ๋ฅผ ๋ค๋ฃน๋๋ค.
YOLO11 ์ฌ์ฉ์ ์ง์ ๋ชจ๋ธ๋ก ๋ฐ์ด๋ฉ ๋ฐ์ค ์์ธก ์ป๊ธฐ
์ด์: ์ฌ์ฉ์ ์ง์ YOLO11 ๋ชจ๋ธ๋ก ์์ธก์ ์คํํ ๋ ๊ฒฝ๊ณ ์์ ์ขํ์ ํ์ ๋ฐ ์๊ฐํ์ ๋ฌธ์ ๊ฐ ์์ต๋๋ค.
์๋ฃจ์ :
- ์ขํ ํ์: YOLO11 ์ ์ ๋ ํฝ์ ๊ฐ์ผ๋ก ๋ฐ์ด๋ฉ ๋ฐ์ค ์ขํ๋ฅผ ์ ๊ณตํฉ๋๋ค. ์ด๋ฅผ ์๋ ์ขํ(0~1 ๋ฒ์)๋ก ๋ณํํ๋ ค๋ฉด ์ด๋ฏธ์ง ํฌ๊ธฐ๋ก ๋๋์ด์ผ ํฉ๋๋ค. ์๋ฅผ ๋ค์ด ์ด๋ฏธ์ง ํฌ๊ธฐ๊ฐ 640x640์ด๋ผ๊ณ ๊ฐ์ ํด ๋ณด๊ฒ ์ต๋๋ค. ๊ทธ๋ฌ๋ฉด ๋ค์์ ์ํํฉ๋๋ค:
# Convert absolute coordinates to relative coordinates
x1 = x1 / 640 # Divide x-coordinates by image width
x2 = x2 / 640
y1 = y1 / 640 # Divide y-coordinates by image height
y2 = y2 / 640
- ํ์ผ ์ด๋ฆ: ์์ธก ๋์ ์ด๋ฏธ์ง์ ํ์ผ ์ด๋ฆ์ ์ป์ผ๋ ค๋ฉด ์์ธก ๋ฃจํ ๋ด์ ๊ฒฐ๊ณผ ๊ฐ์ฒด์์ ์ง์ ์ด๋ฏธ์ง ํ์ผ ๊ฒฝ๋ก์ ์ก์ธ์คํฉ๋๋ค.
YOLO11 ์์ธก์์ ๊ฐ์ฒด ํํฐ๋งํ๊ธฐ
์ด์: Ultralytics ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ฌ์ฉํ์ฌ YOLO11 ์ ์คํํ ๋ ์์ธก ๊ฒฐ๊ณผ์์ ํน์ ๊ฐ์ฒด๋ง ํํฐ๋งํ๊ณ ํ์ํ๋ ๋ฐฉ๋ฒ์ ๋ํ ๋ฌธ์ ๊ฐ ๋ฐ์ํ์ต๋๋ค.
ํด๊ฒฐ ๋ฐฉ๋ฒ: ํน์ ํด๋์ค๋ฅผ ๊ฐ์งํ๋ ค๋ฉด classes ์ธ์๋ฅผ ์ฌ์ฉํ์ฌ ์ถ๋ ฅ์ ํฌํจํ ํด๋์ค๋ฅผ ์ง์ ํฉ๋๋ค. ์๋ฅผ ๋ค์ด, ์๋์ฐจ๋ง ๊ฐ์งํ๋ ค๋ฉด('cars'์ ํด๋์ค ์ธ๋ฑ์ค๊ฐ 2๋ผ๊ณ ๊ฐ์ ):
์ ์ ๋ฐ๋ ์งํ ์ดํด YOLO11
์ด์: ๋ฐ์ค ์ ๋ฐ๋, ๋ง์คํฌ ์ ๋ฐ๋, ํผ๋ ํ๋ ฌ ์ ๋ฐ๋์ ์ฐจ์ด์ ๋ํ ํผ๋ YOLO11.
์๋ฃจ์ : ๋ฐ์ค ์ ๋ฐ๋๋ IoU(Intersection over Union)๋ฅผ ๋ฉํธ๋ฆญ์ผ๋ก ์ฌ์ฉํ์ฌ ์ค์ ๊ธฐ์ค๊ฐ ๋ฐ์ค์ ๋น๊ตํ ์์ธก๋ ๊ฒฝ๊ณ ๋ฐ์ค์ ์ ํ๋๋ฅผ ์ธก์ ํฉ๋๋ค. ๋ง์คํฌ ์ ๋ฐ๋๋ ํฝ์ ๋จ์ ๊ฐ์ฒด ๋ถ๋ฅ์์ ์์ธก๋ ์ธ๋ถํ ๋ง์คํฌ์ ๊ธฐ์ค์ ๋ง์คํฌ ๊ฐ์ ์ผ์น๋๋ฅผ ํ๊ฐํฉ๋๋ค. ๋ฐ๋ฉด ํผ๋ ํ๋ ฌ ์ ๋ฐ๋๋ ๋ชจ๋ ํด๋์ค์ ๊ฑธ์น ์ ๋ฐ์ ์ธ ๋ถ๋ฅ ์ ํ๋์ ์ด์ ์ ๋ง์ถ๋ฉฐ ์์ธก์ ๊ธฐํํ์ ์ ํ๋๋ ๊ณ ๋ คํ์ง ์์ต๋๋ค. ๊ฒฝ๊ณ ์์๋ ํด๋์ค ์์ธก์ด ํ๋ ธ๋๋ผ๋ ๊ธฐํํ์ ์ผ๋ก ์ ํํ ์ ์์ผ๋ฉฐ(์ ๋ต), ๋ฐ๋ผ์ ์์ ์ ๋ฐ๋์ ํผ๋ ํ๋ ฌ ์ ๋ฐ๋ ๊ฐ์ ์ฐจ์ด๊ฐ ๋ฐ์ํ ์ ์๋ค๋ ์ ์ ์ ์ํด์ผ ํฉ๋๋ค. ์ด๋ฌํ ๋ฉํธ๋ฆญ์ ๋ชจ๋ธ ์ฑ๋ฅ์ ๋ค์ํ ์ธก๋ฉด์ ํ๊ฐํ๋ฉฐ, ๋ค์ํ ์์ ์์ ๋ค์ํ ํ๊ฐ ๋ฉํธ๋ฆญ์ด ํ์ํ๋ค๋ ์ ์ ๋ฐ์ํฉ๋๋ค.
์์ ๊ฐ์ฒด ์น์ ์ถ์ถ YOLO11
์ด์: YOLO11 ์์ ๊ฐ์ง๋ ๋ฌผ์ฒด์ ๊ธธ์ด์ ๋์ด๋ฅผ ๊ฒ์ํ๊ธฐ ์ด๋ ค์, ํนํ ์ด๋ฏธ์ง์์ ์ฌ๋ฌ ๊ฐ์ ๋ฌผ์ฒด๊ฐ ๊ฐ์ง๋ ๊ฒฝ์ฐ.
ํด๊ฒฐ ๋ฐฉ๋ฒ: ๋ฐ์ด๋ฉ ๋ฐ์ค ์น์๋ฅผ ๊ฒ์ํ๋ ค๋ฉด ๋จผ์ Ultralytics YOLO11 ๋ชจ๋ธ์ ์ฌ์ฉํ์ฌ ์ด๋ฏธ์ง์ ๊ฐ์ฒด๋ฅผ ์์ธกํฉ๋๋ค. ๊ทธ๋ฐ ๋ค์ ์์ธก ๊ฒฐ๊ณผ์์ ๋ฐ์ด๋ฉ ๋ฐ์ค์ ๋๋น์ ๋์ด ์ ๋ณด๋ฅผ ์ถ์ถํฉ๋๋ค.
from ultralytics import YOLO
# Load a pre-trained YOLO11 model
model = YOLO("yolo11n.pt")
# Specify the source image
source = "https://ultralytics.com/images/bus.jpg"
# Make predictions
results = model.predict(source, save=True, imgsz=320, conf=0.5)
# Extract bounding box dimensions
boxes = results[0].boxes.xywh.cpu()
for box in boxes:
x, y, w, h = box
print(f"Width of Box: {w}, Height of Box: {h}")
๋ฐฐํฌ ๊ณผ์
GPU ๋ฐฐํฌ ๋ฌธ์
์ด์: ๋ค์คGPU ํ๊ฒฝ์์ ๋ชจ๋ธ์ ๋ฐฐํฌํ๋ฉด ์๊ธฐ์น ์์ ๋ฉ๋ชจ๋ฆฌ ์ฌ์ฉ, GPU ๊ฐ ์ผ๊ด๋์ง ์์ ๊ฒฐ๊ณผ ๋ฑ๊ณผ ๊ฐ์ ์๊ธฐ์น ์์ ๋์์ด ๋ฐ์ํ ์ ์์ต๋๋ค.
ํด๊ฒฐ ๋ฐฉ๋ฒ: ๊ธฐ๋ณธ GPU ์ด๊ธฐํ๋ฅผ ํ์ธํฉ๋๋ค. PyTorch ์ ๊ฐ์ ์ผ๋ถ ํ๋ ์์ํฌ๋ ์ง์ ๋ GPU๋ก ์ ํํ๊ธฐ ์ ์ ๊ธฐ๋ณธ GPU ์์ CUDA ์์ ์ ์ด๊ธฐํํ ์ ์์ต๋๋ค. ์๊ธฐ์น ์์ ๊ธฐ๋ณธ ์ด๊ธฐํ๋ฅผ ์ฐํํ๋ ค๋ฉด ๋ฐฐํฌ ๋ฐ ์์ธก ์ค์ GPU ์ ์ง์ ์ง์ ํ์ธ์. ๊ทธ๋ฐ ๋ค์ GPU ์ฌ์ฉ๋ฅ ๋ฐ ๋ฉ๋ชจ๋ฆฌ ์ฌ์ฉ๋์ ๋ชจ๋ํฐ๋งํ๋ ๋๊ตฌ๋ฅผ ์ฌ์ฉํ์ฌ ์ค์๊ฐ์ผ๋ก ์ด์ ์งํ๋ฅผ ํ์ ํ์ธ์. ๋ํ ์ต์ ๋ฒ์ ์ ํ๋ ์์ํฌ ๋๋ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ฌ์ฉํ๊ณ ์๋์ง ํ์ธํ์ธ์.
๋ชจ๋ธ ๋ณํ/๋ด๋ณด๋ด๊ธฐ ๋ฌธ์
๋ฌธ์ : ๋จธ์ ๋ฌ๋ ๋ชจ๋ธ์ ๋ค๋ฅธ ํ์์ด๋ ํ๋ซํผ์ผ๋ก ๋ณํํ๊ฑฐ๋ ๋ด๋ณด๋ด๋ ๊ณผ์ ์์ ์ฌ์ฉ์์๊ฒ ์ค๋ฅ๋ ์๊ธฐ์น ์์ ๋์์ด ๋ฐ์ํ ์ ์์ต๋๋ค.
์๋ฃจ์ :
-
ํธํ์ฑ ํ์ธ: ์๋ก ํธํ๋๋ ๋ผ์ด๋ธ๋ฌ๋ฆฌ ๋ฐ ํ๋ ์์ํฌ ๋ฒ์ ์ ์ฌ์ฉํ๊ณ ์๋์ง ํ์ธํ์ธ์. ๋ฒ์ ์ด ์ผ์นํ์ง ์์ผ๋ฉด ๋ณํ ์ค์ ์๊ธฐ์น ์์ ์ค๋ฅ๊ฐ ๋ฐ์ํ ์ ์์ต๋๋ค.
-
ํ๊ฒฝ ์ฌ์ค์ : Jupyter ๋๋ Colab๊ณผ ๊ฐ์ ๋ํํ ํ๊ฒฝ์ ์ฌ์ฉํ๋ ๊ฒฝ์ฐ ์ค์ํ ๋ณ๊ฒฝ์ด๋ ์ค์น๋ฅผ ์ํํ ํ์๋ ํ๊ฒฝ์ ๋ค์ ์์ํ๋ ๊ฒ์ด ์ข์ต๋๋ค. ์๋ก ์์ํ๋ฉด ๊ทผ๋ณธ์ ์ธ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ ์ ์๋ ๊ฒฝ์ฐ๊ฐ ์์ต๋๋ค.
-
๊ณต์ ๋ฌธ์: ๋ณํ์ ์ฌ์ฉ ์ค์ธ ๋๊ตฌ ๋๋ ๋ผ์ด๋ธ๋ฌ๋ฆฌ์ ๊ณต์ ๋ฌธ์๋ฅผ ํญ์ ์ฐธ์กฐํ์ธ์. ์ฌ๊ธฐ์๋ ๋ชจ๋ธ ๋ด๋ณด๋ด๊ธฐ์ ๋ํ ๊ตฌ์ฒด์ ์ธ ์ง์นจ๊ณผ ๋ชจ๋ฒ ์ฌ๋ก๊ฐ ํฌํจ๋์ด ์๋ ๊ฒฝ์ฐ๊ฐ ๋ง์ต๋๋ค.
-
์ปค๋ฎค๋ํฐ ์ง์: ๋ผ์ด๋ธ๋ฌ๋ฆฌ ๋๋ ํ๋ ์์ํฌ์ ๊ณต์ ๋ฆฌํฌ์งํ ๋ฆฌ์์ ๋ค๋ฅธ ์ฌ์ฉ์๊ฐ ๋ณด๊ณ ํ ์ ์ฌํ ๋ฌธ์ ๋ฅผ ํ์ธํ์ธ์. ์ ์ง ๊ด๋ฆฌ์๋ ์ปค๋ฎค๋ํฐ์์ ํ ๋ก ์ค๋ ๋์์ ํด๊ฒฐ์ฑ ์ด๋ ํด๊ฒฐ ๋ฐฉ๋ฒ์ ์ ๊ณตํ์ ์๋ ์์ต๋๋ค.
-
์ ๊ธฐ์ ์ผ๋ก ์ ๋ฐ์ดํธํฉ๋๋ค: ์ต์ ๋ฒ์ ์ ๋๊ตฌ ๋๋ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ฌ์ฉํ๊ณ ์๋์ง ํ์ธํ์ธ์. ๊ฐ๋ฐ์๋ ์๋ ค์ง ๋ฒ๊ทธ๋ฅผ ์์ ํ๊ฑฐ๋ ๊ธฐ๋ฅ์ ๊ฐ์ ํ๋ ์ ๋ฐ์ดํธ๋ฅผ ์์ฃผ ๋ฆด๋ฆฌ์คํฉ๋๋ค.
-
์ ์ง์ ์ผ๋ก ํ ์คํธํ์ธ์: ์ ์ฒด ๋ณํ์ ์ํํ๊ธฐ ์ ์ ๋ ์์ ๋ชจ๋ธ์ด๋ ๋ฐ์ดํฐ ์ธํธ๋ก ํ๋ก์ธ์ค๋ฅผ ํ ์คํธํ์ฌ ์ ์ฌ์ ์ธ ๋ฌธ์ ๋ฅผ ์กฐ๊ธฐ์ ํ์ ํ์ธ์.
์ปค๋ฎค๋ํฐ ๋ฐ ์ง์
๊ฐ์ ์๊ฐ์ ๊ฐ์ง ์ฌ๋๋ค๋ก ๊ตฌ์ฑ๋ ์ปค๋ฎค๋ํฐ์ ์ฐธ์ฌํ๋ฉด YOLO11 ์์ ์์ ํ๋ ๊ฒฝํ๊ณผ ์ฑ๊ณต์ ํฌ๊ฒ ํฅ์์ํฌ ์ ์์ต๋๋ค. ๋ค์์ ๋์์ด ๋ ๋งํ ๋ช ๊ฐ์ง ์ฑ๋๊ณผ ๋ฆฌ์์ค์ ๋๋ค.
๋์์ ๋ฐ์ ์ ์๋ ํฌ๋ผ ๋ฐ ์ฑ๋
GitHub ์ด์: YOLO11 ๋ฆฌํฌ์งํ ๋ฆฌ์๋ ์ง๋ฌธํ๊ณ , ๋ฒ๊ทธ๋ฅผ ์ ๊ณ ํ๊ณ , ์๋ก์ด ๊ธฐ๋ฅ์ ์ ์ํ ์ ์๋ ์ด์ ํญ์ด ์์ต๋๋ค. ์ปค๋ฎค๋ํฐ์ ๊ด๋ฆฌ์๊ฐ ํ๋ฐํ๊ฒ ํ๋ํ๊ณ ์์ผ๋ฉฐ ํน์ ๋ฌธ์ ์ ๋ํ ๋์์ ๋ฐ์ ์ ์๋ ์ข์ ๊ณณ์ ๋๋ค.
Ultralytics ๋์ค์ฝ๋ ์๋ฒ: Ultralytics ๋ค๋ฅธ ์ฌ์ฉ์ ๋ฐ ๊ฐ๋ฐ์์ ์ํตํ ์ ์๋ ๋์ค์ฝ๋ ์๋ฒ๊ฐ ์์ต๋๋ค.
๊ณต์ ๋ฌธ์ ๋ฐ ๋ฆฌ์์ค
Ultralytics YOLO11 ๋ฌธ์: ๊ณต์ ๋ฌธ์์์๋ ์ค์น, ์ฌ์ฉ ๋ฐ ๋ฌธ์ ํด๊ฒฐ์ ๋ํ ๊ฐ์ด๋์ ํจ๊ป YOLO11 ์ ๋ํ ํฌ๊ด์ ์ธ ๊ฐ์๋ฅผ ์ ๊ณตํฉ๋๋ค.
์ด๋ฌํ ๋ฆฌ์์ค๋ YOLO11 ํ๋ก์ ํธ์ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๊ณ ๊ฐ์ ํ๋ฉฐ YOLO11 ์ปค๋ฎค๋ํฐ์ ๋ค๋ฅธ ์ฌ๋๋ค๊ณผ ์ํตํ ์ ์๋ ํํํ ํ ๋๊ฐ ๋ ๊ฒ์ ๋๋ค.
๊ฒฐ๋ก
๋ฌธ์ ํด๊ฒฐ์ ๋ชจ๋ ๊ฐ๋ฐ ํ๋ก์ธ์ค์์ ํ์์ ์ธ ๋ถ๋ถ์ด๋ฉฐ, ์ฌ๋ฐ๋ฅธ ์ง์์ ๊ฐ์ถ๋ฉด ๋ฌธ์ ํด๊ฒฐ์ ์์๋๋ ์๊ฐ๊ณผ ๋ ธ๋ ฅ์ ํฌ๊ฒ ์ค์ผ ์ ์์ต๋๋ค. ์ด ๊ฐ์ด๋๋ Ultralytics ์ํ๊ณ ๋ด์์ YOLO11 ๋ชจ๋ธ ์ฌ์ฉ์๊ฐ ์ง๋ฉดํ๋ ๊ฐ์ฅ ์ผ๋ฐ์ ์ธ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋ ๊ฒ์ ๋ชฉํ๋ก ์์ฑ๋์์ต๋๋ค. ์ด๋ฌํ ์ผ๋ฐ์ ์ธ ๋ฌธ์ ๋ฅผ ์ดํดํ๊ณ ํด๊ฒฐํจ์ผ๋ก์จ ํ๋ก์ ํธ๋ฅผ ๋ ์ํํ๊ฒ ์งํํ๊ณ ์ปดํจํฐ ๋น์ ์์ ์์ ๋ ๋์ ๊ฒฐ๊ณผ๋ฅผ ์ป์ ์ ์์ต๋๋ค.
Ultralytics ์ปค๋ฎค๋ํฐ๋ ๊ท์คํ ๋ฆฌ์์ค์์ ๊ธฐ์ตํ์ธ์. ๋๋ฃ ๊ฐ๋ฐ์ ๋ฐ ์ ๋ฌธ๊ฐ์ ์ํตํ๋ฉด ํ์ค ๋ฌธ์์์ ๋ค๋ฃจ์ง ์์ ์ถ๊ฐ์ ์ธ ์ธ์ฌ์ดํธ์ ์๋ฃจ์ ์ ์ป์ ์ ์์ต๋๋ค. ํญ์ ๋ฐฐ์ฐ๊ณ , ์คํํ๊ณ , ๊ฒฝํ์ ๊ณต์ ํ์ฌ ์ปค๋ฎค๋ํฐ์ ์ง๋จ ์ง์์ ๊ธฐ์ฌํ์ธ์.
ํ๋ณตํ ๋ฌธ์ ํด๊ฒฐ!
์์ฃผ ๋ฌป๋ ์ง๋ฌธ
์ค์น ์ค๋ฅ๋ ์ด๋ป๊ฒ ํด๊ฒฐํ๋์ YOLO11?
์ค์น ์ค๋ฅ๋ ํธํ์ฑ ๋ฌธ์ ๋ ์ข ์์ฑ ๋๋ฝ์ผ๋ก ์ธํด ๋ฐ์ํ๋ ๊ฒฝ์ฐ๊ฐ ๋ง์ต๋๋ค. Python 3.8 ์ด์์ ์ฌ์ฉํ๊ณ PyTorch 1.8 ์ด์์ ์ค์นํ๋์ง ํ์ธํ์ธ์. ์ถฉ๋์ ํผํ๋ ค๋ฉด ๊ฐ์ ํ๊ฒฝ์ ์ฌ์ฉํ๋ ๊ฒ์ด ์ข์ต๋๋ค. ๋จ๊ณ๋ณ ์ค์น ๊ฐ์ด๋๋ ๊ณต์ ์ค์น ๊ฐ์ด๋๋ฅผ ์ฐธ์กฐํ์ธ์. ๊ฐ์ ธ์ค๊ธฐ ์ค๋ฅ๊ฐ ๋ฐ์ํ๋ฉด ์๋ก ์ค์นํ๊ฑฐ๋ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ต์ ๋ฒ์ ์ผ๋ก ์ ๋ฐ์ดํธํ์ธ์.
YOLO11 ๋ชจ๋ธ ๊ต์ก์ด ๋จ์ผ GPU ์์ ๋๋ฆฐ ์ด์ ๋ ๋ฌด์์ธ๊ฐ์?
๋ฐฐ์น ํฌ๊ธฐ๊ฐ ํฌ๊ฑฐ๋ ๋ฉ๋ชจ๋ฆฌ๊ฐ ๋ถ์กฑํ์ฌ ๋จ์ผ GPU ์์ ํ๋ จํ๋ ์๋๊ฐ ๋๋ ค์ง ์ ์์ต๋๋ค. ํ๋ จ ์๋๋ฅผ ๋์ด๋ ค๋ฉด ์ฌ๋ฌ ๊ฐ์ GPU๋ฅผ ์ฌ์ฉํ์ธ์. ์์คํ
์ ์ฌ๋ฌ ๊ฐ์ GPU๋ฅผ ์ฌ์ฉํ ์ ์๋์ง ํ์ธํ๊ณ .yaml
๊ตฌ์ฑ ํ์ผ์ ์ฌ์ฉํ์ฌ GPU ์๋ฅผ ์ง์ ํ ์ ์์ต๋๋ค, gpus: 4
. ๋ฉ๋ชจ๋ฆฌ ์ ํ์ ์ด๊ณผํ์ง ์๊ณ GPU๋ฅผ ์ต๋ํ ํ์ฉํ๋ ค๋ฉด ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ์ ์ ํ ๋๋ฆฝ๋๋ค. ๋ช
๋ น ์์:
YOLO11 ๋ชจ๋ธ์ด GPU ์์ ๊ต์ก์ ๋ฐ๊ณ ์๋์ง ํ์ธํ๋ ค๋ฉด ์ด๋ป๊ฒ ํด์ผ ํ๋์?
ํธ๋ ์ด๋ ๋ก๊ทธ์์ 'device' ๊ฐ์ด 'null'๋ก ํ์๋๋ฉด ์ผ๋ฐ์ ์ผ๋ก ํธ๋ ์ด๋ ํ๋ก์ธ์ค๊ฐ ์ฌ์ฉ ๊ฐ๋ฅํ GPU ์ ์๋์ผ๋ก ์ฌ์ฉํ๋๋ก ์ค์ ๋์ด ์์์ ์๋ฏธํฉ๋๋ค. ํน์ GPU ์ ๋ช
์์ ์ผ๋ก ํ ๋นํ๋ ค๋ฉด, ์ฌ๋ฌ๋ถ์ .yaml
๊ตฌ์ฑ ํ์ผ์ ์ถ๊ฐํฉ๋๋ค. ์๋ฅผ ๋ค์ด
์ด๋ ๊ฒ ํ๋ฉด ๊ต์ก ํ๋ก์ธ์ค๊ฐ ์ฒซ ๋ฒ์งธ GPU ๋ก ์ค์ ๋ฉ๋๋ค. ๋ฅผ ์ฐธ์กฐํ์ญ์์ค. nvidia-smi
๋ช
๋ น์ ์คํํ์ฌ CUDA ์ค์ ์ ํ์ธํฉ๋๋ค.
YOLO11 ๋ชจ๋ธ ๊ต์ก ์งํ ์ํฉ์ ๋ชจ๋ํฐ๋งํ๊ณ ์ถ์ ํ๋ ค๋ฉด ์ด๋ป๊ฒ ํด์ผ ํ๋์?
๊ต์ก ์งํ ์ํฉ์ ์ถ์ ํ๊ณ ์๊ฐํํ๋ ๊ฒ์ TensorBoard์ ๊ฐ์ ๋๊ตฌ๋ฅผ ํตํด ํจ์จ์ ์ผ๋ก ๊ด๋ฆฌํ ์ ์์ต๋๋ค, Comet, Ultralytics ํ๋ธ์ ๊ฐ์ ๋๊ตฌ๋ฅผ ํตํด ํจ์จ์ ์ผ๋ก ๊ด๋ฆฌํ ์ ์์ต๋๋ค. ์ด๋ฌํ ๋๊ตฌ๋ฅผ ์ฌ์ฉํ๋ฉด ์์ค, ์ ํ๋, ๋ฆฌ์ฝ ๋ฐ mAP์ ๊ฐ์ ์งํ๋ฅผ ๊ธฐ๋กํ๊ณ ์๊ฐํํ ์ ์์ต๋๋ค. ์ด๋ฌํ ์งํ๋ฅผ ๊ธฐ๋ฐ์ผ๋ก ์กฐ๊ธฐ ์ค์ง๋ฅผ ๊ตฌํํ๋ฉด ๋ ๋์ ํ๋ จ ๊ฒฐ๊ณผ๋ฅผ ๋ฌ์ฑํ๋ ๋ฐ ๋์์ด ๋ ์ ์์ต๋๋ค.
YOLO11 ์์ ๋ด ๋ฐ์ดํฐ ์ธํธ ํ์์ ์ธ์ํ์ง ๋ชปํ๋ ๊ฒฝ์ฐ ์ด๋ป๊ฒ ํด์ผ ํ๋์?
๋ฐ์ดํฐ ์ธํธ์ ๋ ์ด๋ธ์ด ์์ ํ์์ ์ค์ํ๋์ง ํ์ธํฉ๋๋ค. ์ฃผ์์ด ์ ํํ๊ณ ๊ณ ํ์ง์ธ์ง ํ์ธํฉ๋๋ค. ๋ฌธ์ ๊ฐ ๋ฐ์ํ๋ฉด ๋ฐ์ดํฐ ์์ง ๋ฐ ์ด๋ ธํ ์ด์ ๊ฐ์ด๋์์ ๋ชจ๋ฒ ์ฌ๋ก๋ฅผ ์ฐธ์กฐํ์ธ์. ๋ฐ์ดํฐ ์งํฉ์ ๋ํ ์์ธํ ์ง์นจ์ ๋ฌธ์์์ ๋ฐ์ดํฐ ์งํฉ ์น์ ์ ์ฐธ์กฐํ์ธ์.