๋ค์์ ๋ํ ์ข ํฉ ๊ฐ์ด๋ Ultralytics YOLOv5
Ultralytics' YOLOv5๐ ๋ฌธ์ํ! YOLOv5 ์ ๋ค์ฏ ๋ฒ์งธ ๋ฒ์ ์ธ ํ์ ์ ์ธ "You Only Look Once"๋ฅผ ์๊ฐํฉ๋๋ค. ๋ฌผ์ฒด ๊ฐ์ง ๋ชจ๋ธ์ ๊ณ ์, ๊ณ ์ ๋ฐ ๊ฒฐ๊ณผ๋ฅผ ์ค์๊ฐ์ผ๋ก ์ ๊ณตํ๋๋ก ์ค๊ณ๋์์ต๋๋ค.
PyTorch ์ ๊ตฌ์ถ๋ ์ด ๊ฐ๋ ฅํ ๋ฅ ๋ฌ๋ ํ๋ ์์ํฌ๋ ๋ค์ฉ๋์ฑ, ์ฌ์ฉ ํธ์์ฑ, ๊ณ ์ฑ๋ฅ์ผ๋ก ํฐ ์ธ๊ธฐ๋ฅผ ์ป๊ณ ์์ต๋๋ค. ์ด ๋ฌธ์๋ ์ค์น ๊ณผ์ ์ ์๋ดํ๊ณ , ๋ชจ๋ธ์ ์ํคํ ์ฒ์ ๋์์ค๋ฅผ ์ค๋ช ํ๊ณ , ๋ค์ํ ์ฌ์ฉ ์ฌ๋ก๋ฅผ ๋ณด์ฌ์ฃผ๋ฉฐ, ์ผ๋ จ์ ์์ธํ ํํ ๋ฆฌ์ผ์ ์ ๊ณตํฉ๋๋ค. ์ด๋ฌํ ๋ฆฌ์์ค๋ฅผ ํตํด YOLOv5 ์ ์ ์ฌ๋ ฅ์ ์ต๋ํ ํ์ฉํ ์ ์์ต๋๋ค. ์ปดํจํฐ ๋น์ ํ๋ก์ ํธ์ ์ฐธ์ฌํ์ธ์. ์์ํด๋ณด์!
ํ์ ๋ฐ ํ์ต
๋ค์์ YOLOv5 ์ ๋ค์ํ ์ธก๋ฉด์ ์๋ดํ๋ ์ข ํฉ์ ์ธ ํํ ๋ฆฌ์ผ ๋ชจ์์ ๋๋ค.
- ์ฌ์ฉ์ ์ง์ ๋ฐ์ดํฐ ํ๋ จ ๐ ๊ถ์ฅ: ์ฌ์ฉ์ ์ง์ ๋ฐ์ดํฐ ์งํฉ์์ YOLOv5 ๋ชจ๋ธ์ ํ๋ จํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์.
- ์ต์์ ํธ๋ ์ด๋ ๊ฒฐ๊ณผ๋ฅผ ์ํ ํ โ๏ธ: ๋ชจ๋ธ ํธ๋ ์ด๋ ํ๋ก์ธ์ค๋ฅผ ์ต์ ํํ๊ธฐ ์ํ ์ค์ฉ์ ์ธ ํ์ ์์๋ณด์ธ์.
- ๋ฉํฐGPU ๊ต์ก: ๋ฉํฐ GPU๋ฅผ ํ์ฉํ์ฌ ํธ๋ ์ด๋์ ๊ฐ์ํํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์.
- PyTorch Hub ๐ ์ ๊ท: PyTorch Hub๋ฅผ ํตํด ์ฌ์ ํ์ต๋ ๋ชจ๋ธ์ ๋ก๋ํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์.
- TFLite, ONNX, CoreML, TensorRT ๋ด๋ณด๋ด๊ธฐ ๐: ๋ชจ๋ธ์ ๋ค์ํ ํ์์ผ๋ก ๋ด๋ณด๋ด๋ ๋ฐฉ๋ฒ์ ์ดํดํฉ๋๋ค.
- ํ ์คํธ ์๊ฐ ์ฆ๊ฐ(TTA): TTA๋ฅผ ์ฌ์ฉํ์ฌ ๋ชจ๋ธ์ ์์ธก ์ ํ๋๋ฅผ ๊ฐ์ ํ๋ ๋ฐฉ๋ฒ์ ์ดํด๋ณด์ธ์.
- ๋ชจ๋ธ ์์๋ธ: ์ฌ๋ฌ ๋ชจ๋ธ์ ๊ฒฐํฉํ์ฌ ์ฑ๋ฅ์ ํฅ์์ํค๋ ์ ๋ต์ ์์๋ณด์ธ์.
- ๋ชจ๋ธ ๊ฐ์ง์น๊ธฐ/์คํ์ค์ํฐ: ๊ฐ์ง ์น๊ธฐ ๋ฐ ํฌ์์ฑ ๊ฐ๋ ๊ณผ ๋ณด๋ค ํจ์จ์ ์ธ ๋ชจ๋ธ์ ๋ง๋๋ ๋ฐฉ๋ฒ์ ์ดํดํฉ๋๋ค.
- ํ์ดํผํ๋ผ๋ฏธํฐ ์งํ: ๋ชจ๋ธ ์ฑ๋ฅ ํฅ์์ ์ํ ์๋ํ๋ ํ์ดํผํ๋ผ๋ฏธํฐ ํ๋ ํ๋ก์ธ์ค์ ๋ํด ์์๋ณด์ธ์.
- ๊ณ ์ ๋ ์ด์ด๋ฅผ ์ฌ์ฉํ ์ ์ด ํ์ต: YOLOv5 ์์ ๋ ์ด์ด๋ฅผ ๊ณ ์ ํ์ฌ ์ ์ด ํ์ต์ ๊ตฌํํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์.
- ์ํคํ ์ฒ ์์ฝ ๐ YOLOv5 ๋ชจ๋ธ์ ๊ตฌ์กฐ์ ์ธ๋ถ ์ฌํญ์ ์์ธํ ์ดํด๋ณด์ธ์.
- Roboflow ์ ๋ํด ์์๋ณด์ธ์: ๋ฐ์ดํฐ ์ธํธ ๊ด๋ฆฌ, ๋ผ๋ฒจ๋ง ๋ฐ ๋ฅ๋์ ํ์ต์ ์ํด Roboflow ์ ํ์ฉํ๋ ๋ฐฉ๋ฒ์ ์ดํดํฉ๋๋ค.
- ClearML ๋ก๊น ๐ ๋ชจ๋ธ ํธ๋ ์ด๋ ์ค ํจ์จ์ ์ธ ๋ก๊น ์ ์ํด ClearML ์ ํตํฉํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์.
- YOLOv5 Neural Magic Neural Magic ๋ฅ์คํ์ค๋ฅผ ์ฌ์ฉํ์ฌ ๋ชจ๋ธ์ ๊ฐ์ง์น๊ธฐํ๊ณ ์ ๋ํํ๋ ๋ฐฉ๋ฒ์ ์์๋ณด์ธ์. YOLOv5
- Comet ๋ก๊น ๐ ์ ๊ท: ๋ชจ๋ธ ํธ๋ ์ด๋ ๋ก๊น ์ ๊ฐ์ ํ๊ธฐ ์ํด Comet ์ ํ์ฉํ๋ ๋ฐฉ๋ฒ์ ์ดํด๋ณด์ธ์.
์ง์ ํ๊ฒฝ
Ultralytics ๋ ๋ฐ๋ก ์ฌ์ฉํ ์ ์๋ ๋ค์ํ ํ๊ฒฝ์ ์ ๊ณตํ๋ฉฐ, ๊ฐ ํ๊ฒฝ์๋ ๋ค์๊ณผ ๊ฐ์ ํ์ ์ข ์์ฑ์ด ์ฌ์ ์ค์น๋์ด ์์ต๋๋ค. CUDA, CUDNN, Python, ๋ฐ PyTorch์ ๊ฐ์ ํ์ ์ข ์ ์์๋ฅผ ์ค์นํ์ฌ ํ๋ก์ ํธ๋ฅผ ์์ํ ์ ์์ต๋๋ค.
- ๋ฌด๋ฃ GPU ๋ ธํธ๋ถ:
- Google Cloud: GCP ๋น ๋ฅธ ์์ ๊ฐ์ด๋
- Amazon: AWS ๋น ๋ฅธ ์์ ๊ฐ์ด๋
- Azure: AzureML ๋น ๋ฅธ ์์ ๊ฐ์ด๋
- Docker: Docker ๋น ๋ฅธ ์์ ๊ฐ์ด๋
ํ๋ก์ ํธ ์ํ
์ด ๋ฐฐ์ง๋ ๋ชจ๋ YOLOv5 GitHub Actions ์ง์์ ํตํฉ(CI) ํ ์คํธ๊ฐ ์ฑ๊ณต์ ์ผ๋ก ํต๊ณผ๋์์์ ๋ํ๋ ๋๋ค. ์ด๋ฌํ CI ํ ์คํธ๋ ๊ต์ก, ๊ฒ์ฆ, ์ถ๋ก , ๋ด๋ณด๋ด๊ธฐ ๋ฐ ๋ฒค์น๋งํฌ ๋ฑ ๋ค์ํ ์ฃผ์ ์ธก๋ฉด์์ YOLOv5 ์ ๊ธฐ๋ฅ๊ณผ ์ฑ๋ฅ์ ์๊ฒฉํ๊ฒ ํ์ธํฉ๋๋ค. 24์๊ฐ๋ง๋ค ๊ทธ๋ฆฌ๊ณ ์๋ก์ด ์ปค๋ฐ์ด ์์ ๋๋ง๋ค ํ ์คํธ๋ฅผ ์ํํ์ฌ macOS, Windows ๋ฐ Ubuntu์์ ์ผ๊ด๋๊ณ ์์ ์ ์ธ ์๋์ ๋ณด์ฅํฉ๋๋ค.
์ฐ๊ฒฐ ๋ฐ ๊ธฐ์ฌ
YOLOv5 ์์ ์ฌ์ ์ด ์ธ๋ก์ธ ํ์๋ ์์ต๋๋ค. GitHub์ ํ๊ธฐ์ฐฌ ์ปค๋ฎค๋ํฐ์ ์ฐธ์ฌํ๊ณ , LinkedIn์์ ์ ๋ฌธ๊ฐ๋ค๊ณผ ๊ต๋ฅํ๊ณ , Twitter์์ ๊ฒฐ๊ณผ๋ฅผ ๊ณต์ ํ๊ณ , YouTube์์ ๊ต์ก ๋ฆฌ์์ค๋ฅผ ์ฐพ์๋ณด์ธ์. ๋ ํฅ๋ฏธ๋ก์ด ์ฝํ ์ธ ๋ฅผ ๋ณด๋ ค๋ฉด TikTok๊ณผ BiliBili๋ฅผ ํ๋ก์ฐํ์ธ์.
๊ธฐ์ฌํ๊ณ ์ถ์ผ์ ๊ฐ์? ์ฝ๋ ๊ฐ์ ๋ฐ ๋ฒ๊ทธ ๋ณด๊ณ ๋ถํฐ ๋ฌธ์ ์ ๋ฐ์ดํธ์ ์ด๋ฅด๊ธฐ๊น์ง ๋ชจ๋ ํํ์ ๊ธฐ์ฌ๋ฅผ ํ์ํฉ๋๋ค. ์์ธํ ๋ด์ฉ์ ๊ธฐ์ฌ ๊ฐ์ด๋๋ผ์ธ์ ์ฐธ์กฐํ์ธ์.
์ฌ๋ฌ๋ถ์ ํ์ ์ ์ธ ์ฌ์ฉ๋ฒ์ ๊ธฐ๋ํฉ๋๋ค YOLOv5. ์ปดํจํฐ ๋น์ ํ๋ก์ ํธ์ ๋ฐ์ด๋ค์ด ์คํํ๊ณ ํ์ ํ์ธ์! ๐
์์ฃผ ๋ฌป๋ ์ง๋ฌธ
Ultralytics YOLOv5 ์ ์ฃผ์ ๊ธฐ๋ฅ์ ๋ฌด์์ธ๊ฐ์?
Ultralytics YOLOv5 ๋ ๊ณ ์, ๊ณ ์ ๋ฐ ๋ฌผ์ฒด ๊ฐ์ง ๊ธฐ๋ฅ์ผ๋ก ์ ๋ช ํฉ๋๋ค. ๊ธฐ๋ฐ PyTorch๋ค๋ชฉ์ ์ด๋ฉฐ ์ฌ์ฉ์ ์นํ์ ์ด์ด์ ๋ค์ํ ์ปดํจํฐ ๋น์ ํ๋ก์ ํธ์ ์ ํฉํฉ๋๋ค. ์ฃผ์ ๊ธฐ๋ฅ์ผ๋ก๋ ์ค์๊ฐ ์ถ๋ก , TTA(ํ ์คํธ ์๊ฐ ์ฆ๊ฐ) ๋ฐ ๋ชจ๋ธ ์์๋ธ๊ณผ ๊ฐ์ ์ฌ๋ฌ ํ๋ จ ๊ธฐ๋ฒ ์ง์, TFLite, ONNX, CoreML, TensorRT ๊ณผ ๊ฐ์ ๋ด๋ณด๋ด๊ธฐ ํ์๊ณผ์ ํธํ์ฑ ๋ฑ์ด ์์ต๋๋ค. Ultralytics YOLOv5 ์ ํตํด ํ๋ก์ ํธ๋ฅผ ํฅ์์ํฌ ์ ์๋ ๋ฐฉ๋ฒ์ ๋ํด ์์ธํ ์์๋ณด๋ ค๋ฉด TFLite, ONNX, CoreML, TensorRT ๋ด๋ณด๋ด๊ธฐ ๊ฐ์ด๋๋ฅผ ์ฐธ์กฐํ์ธ์.
๋ด ๋ฐ์ดํฐ ์งํฉ์์ ์ฌ์ฉ์ ์ง์ YOLOv5 ๋ชจ๋ธ์ ํ์ตํ๋ ค๋ฉด ์ด๋ป๊ฒ ํด์ผ ํ๋์?
๋ฐ์ดํฐ ์ธํธ์์ ์ฌ์ฉ์ ์ง์ YOLOv5 ๋ชจ๋ธ์ ํ๋ จํ๋ ค๋ฉด ๋ช ๊ฐ์ง ์ฃผ์ ๋จ๊ณ๊ฐ ํ์ํฉ๋๋ค. ๋จผ์ , ๋ ์ด๋ธ๋ก ์ฃผ์์ ๋ฌ์ ํ์ํ ํ์์ผ๋ก ๋ฐ์ดํฐ ์งํฉ์ ์ค๋นํฉ๋๋ค. ๊ทธ๋ฐ ๋ค์ YOLOv5 ํ๋ จ ๋งค๊ฐ๋ณ์๋ฅผ ๊ตฌ์ฑํ๊ณ ๋ค์์ ์ฌ์ฉํ์ฌ ํ๋ จ ํ๋ก์ธ์ค๋ฅผ ์์ํฉ๋๋ค. train.py
์คํฌ๋ฆฝํธ๋ฅผ ์ฌ์ฉํ์ธ์. ์ด ํ๋ก์ธ์ค์ ๋ํ ์์ธํ ํํ ๋ฆฌ์ผ์ ๋ณด๋ ค๋ฉด ์ฌ์ฉ์ ์ง์ ๋ฐ์ดํฐ ๊ฐ์ด๋ ๊ต์ก. ํน์ ์ฌ์ฉ ์ฌ๋ก์ ๋ง๋ ์ต์ ์ ๊ฒฐ๊ณผ๋ฅผ ์ป์ ์ ์๋๋ก ๋จ๊ณ๋ณ ์ง์นจ์ ์ ๊ณตํฉ๋๋ค.
RCNN๊ณผ ๊ฐ์ ๋ค๋ฅธ ๊ฐ์ฒด ๊ฐ์ง ๋ชจ๋ธ ๋์ Ultralytics YOLOv5 ์ ์ฌ์ฉํด์ผ ํ๋ ์ด์ ๋ ๋ฌด์์ธ๊ฐ์?
Ultralytics YOLOv5 ๋ ์ค์๊ฐ ๋ฌผ์ฒด ๊ฐ์ง์ ์๋์ ์ ํ์ฑ์ด ๋ฐ์ด๋ RCNN๊ณผ ๊ฐ์ ๋ชจ๋ธ๋ณด๋ค ์ ํธ๋ฉ๋๋ค. YOLOv5 ๋ ์ ์ฒด ์ด๋ฏธ์ง๋ฅผ ํ ๋ฒ์ ์ฒ๋ฆฌํ๋ฏ๋ก ์ฌ๋ฌ ๋ฒ์ ํจ์ค๋ฅผ ๊ฑฐ์น๋ RCNN์ ์์ญ ๊ธฐ๋ฐ ์ ๊ทผ ๋ฐฉ์์ ๋นํด ํจ์ฌ ๋น ๋ฆ ๋๋ค. ๋ํ YOLOv5 ์ ๋ค์ํ ๋ด๋ณด๋ด๊ธฐ ํ์๊ณผ ์ํํ๊ฒ ํตํฉ๋๊ณ ๊ด๋ฒ์ํ ๋ฌธ์๊ฐ ์ ๊ณต๋๋ฏ๋ก ์ด๋ณด์์ ์ ๋ฌธ๊ฐ ๋ชจ๋์๊ฒ ํ์ํ ์ ํ์ด ๋ ์ ์์ต๋๋ค. ์ํคํ ์ฒ ์์ฝ์์ ์ํคํ ์ฒ์ ์ฅ์ ์ ๋ํด ์์ธํ ์์๋ณด์ธ์.
๊ต์ก ์ค์ YOLOv5 ๋ชจ๋ธ ์ฑ๋ฅ์ ์ต์ ํํ๋ ค๋ฉด ์ด๋ป๊ฒ ํด์ผ ํ๋์?
YOLOv5 ๋ชจ๋ธ ์ฑ๋ฅ ์ต์ ํ์๋ ๋ค์ํ ํ์ดํผํ๋ผ๋ฏธํฐ๋ฅผ ์กฐ์ ํ๊ณ ๋ฐ์ดํฐ ์ฆ๊ฐ ๋ฐ ์ ์ด ํ์ต๊ณผ ๊ฐ์ ๊ธฐ์ ์ ํตํฉํ๋ ๊ฒ์ด ํฌํจ๋ฉ๋๋ค. Ultralytics ์์๋ ๋ชจ๋ธ ํจ์จ์ฑ ํฅ์์ ์ํ ํ์ดํผํ๋ผ๋ฏธํฐ ์งํ ๋ฐ ๊ฐ์ง์น๊ธฐ/์คํ์ค์ํฐ์ ๋ํ ์ข ํฉ์ ์ธ ๋ฆฌ์์ค๋ฅผ ์ ๊ณตํฉ๋๋ค. ํ์ต ์ค์ ์ต์ ์ ์ฑ๋ฅ์ ๋ฌ์ฑํ๊ธฐ ์ํ ์คํ ๊ฐ๋ฅํ ์ธ์ฌ์ดํธ๋ฅผ ์ ๊ณตํ๋ ์ต์์ ํ์ต ๊ฒฐ๊ณผ๋ฅผ ์ํ ํ ๊ฐ์ด๋์์ ์ค์ฉ์ ์ธ ํ์ ํ์ธํ ์ ์์ต๋๋ค.
YOLOv5 ์ ํ๋ฆฌ์ผ์ด์ ์ ์คํํ๊ธฐ ์ํด ์ง์๋๋ ํ๊ฒฝ์ ๋ฌด์์ธ๊ฐ์?
Ultralytics YOLOv5 ๋ Gradient์ ๋ฌด๋ฃ GPU ๋ ธํธ๋ถ, Google Colab, Kaggle์ ๋ฌผ๋ก Google Cloud, Amazon AWS, Azure์ ๊ฐ์ ์ฃผ์ ํด๋ผ์ฐ๋ ํ๋ซํผ ๋ฑ ๋ค์ํ ํ๊ฒฝ์ ์ง์ํฉ๋๋ค. ํธ๋ฆฌํ ์ค์ ์ ์ํด Docker ์ด๋ฏธ์ง๋ ์ ๊ณต๋ฉ๋๋ค. ์ด๋ฌํ ํ๊ฒฝ ์ค์ ์ ๋ํ ์์ธํ ๊ฐ์ด๋๋ ๊ฐ ํ๋ซํผ์ ๋ํ ๋จ๊ณ๋ณ ์ง์นจ์ด ํฌํจ๋ ์ง์๋๋ ํ๊ฒฝ ์น์ ์ ์ฐธ์กฐํ์ธ์.