์ฝ˜ํ…์ธ ๋กœ ๊ฑด๋„ˆ๋›ฐ๊ธฐ

์„ฑ๋Šฅ ์ง€ํ‘œ ์‹ฌ์ธต ๋ถ„์„

์†Œ๊ฐœ

์„ฑ๋Šฅ ๋ฉ”ํŠธ๋ฆญ์€ ๊ฐ์ฒด ๊ฐ์ง€ ๋ชจ๋ธ์˜ ์ •ํ™•์„ฑ๊ณผ ํšจ์œจ์„ฑ์„ ํ‰๊ฐ€ํ•˜๋Š” ํ•ต์‹ฌ ๋„๊ตฌ์ž…๋‹ˆ๋‹ค. ์ด ์ง€ํ‘œ๋Š” ๋ชจ๋ธ์ด ์ด๋ฏธ์ง€ ๋‚ด์—์„œ ๊ฐ์ฒด๋ฅผ ์–ผ๋งˆ๋‚˜ ํšจ๊ณผ์ ์œผ๋กœ ์‹๋ณ„ํ•˜๊ณ  ์œ„์น˜๋ฅผ ํŒŒ์•…ํ•  ์ˆ˜ ์žˆ๋Š”์ง€๋ฅผ ์•Œ๋ ค์ค๋‹ˆ๋‹ค. ๋˜ํ•œ ์˜คํƒ๊ณผ ์˜คํƒ์— ๋Œ€ํ•œ ๋ชจ๋ธ์˜ ์ฒ˜๋ฆฌ ๋ฐฉ์‹์„ ์ดํ•ดํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋ฉ๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ์ธ์‚ฌ์ดํŠธ๋Š” ๋ชจ๋ธ์˜ ์„ฑ๋Šฅ์„ ํ‰๊ฐ€ํ•˜๊ณ  ํ–ฅ์ƒ์‹œํ‚ค๋Š” ๋ฐ ๋งค์šฐ ์ค‘์š”ํ•ฉ๋‹ˆ๋‹ค. ์ด ๊ฐ€์ด๋“œ์—์„œ๋Š” YOLOv8 ์™€ ๊ด€๋ จ๋œ ๋‹ค์–‘ํ•œ ์„ฑ๋Šฅ ๋ฉ”ํŠธ๋ฆญ๊ณผ ๊ทธ ์ค‘์š”์„ฑ, ๊ทธ๋ฆฌ๊ณ  ์ด๋ฅผ ํ•ด์„ํ•˜๋Š” ๋ฐฉ๋ฒ•์„ ์‚ดํŽด๋ด…๋‹ˆ๋‹ค.



Watch: Ultralytics YOLOv8 ์„ฑ๋Šฅ ์ง€ํ‘œ | MAP, F1 ์ ์ˆ˜, ์ •๋ฐ€๋„, IoU ๋ฐ ์ •ํ™•๋„

๊ฐ์ฒด ๊ฐ์ง€ ๋ฉ”ํŠธ๋ฆญ

YOLOv8 ์—๋งŒ ์ค‘์š”ํ•œ ๊ฒƒ์ด ์•„๋‹ˆ๋ผ ๋‹ค์–‘ํ•œ ๊ฐ์ฒด ๊ฐ์ง€ ๋ชจ๋ธ์— ๊ด‘๋ฒ”์œ„ํ•˜๊ฒŒ ์ ์šฉ๋  ์ˆ˜ ์žˆ๋Š” ๋ช‡ ๊ฐ€์ง€ ๋ฉ”ํŠธ๋ฆญ์— ๋Œ€ํ•ด ๋…ผ์˜ํ•˜๋Š” ๊ฒƒ์œผ๋กœ ์‹œ์ž‘ํ•˜๊ฒ ์Šต๋‹ˆ๋‹ค.

  • ์œ ๋‹ˆ์˜จ์— ๋Œ€ํ•œ ๊ต์ฐจ(IoU): IoU๋Š” ์˜ˆ์ธก๋œ ๋ฐ”์šด๋”ฉ ๋ฐ•์Šค์™€ ์‹ค์ธก ๋ฐ”์šด๋”ฉ ๋ฐ•์Šค ์‚ฌ์ด์˜ ์ค‘์ฒฉ์„ ์ •๋Ÿ‰ํ™”ํ•˜๋Š” ์ธก์ •๊ฐ’์ž…๋‹ˆ๋‹ค. ์ด๋Š” ๊ฐ์ฒด ์ธก์œ„์˜ ์ •ํ™•๋„๋ฅผ ํ‰๊ฐ€ํ•˜๋Š” ๋ฐ ๊ธฐ๋ณธ์ ์ธ ์—ญํ• ์„ ํ•ฉ๋‹ˆ๋‹ค.

  • ํ‰๊ท  ์ •๋ฐ€๋„(AP): AP๋Š” ์ •๋ฐ€๋„-์žฌ์‘๋‹ต ๊ณก์„  ์•„๋ž˜์˜ ์˜์—ญ์„ ๊ณ„์‚ฐํ•˜์—ฌ ๋ชจ๋ธ์˜ ์ •๋ฐ€๋„์™€ ์žฌ์‘๋‹ต ์„ฑ๋Šฅ์„ ์š”์•ฝํ•˜๋Š” ๋‹จ์ผ ๊ฐ’์„ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

  • ํ‰๊ท  ํ‰๊ท  ์ •๋ฐ€๋„(mAP): mAP๋Š” ์—ฌ๋Ÿฌ ๊ฐ์ฒด ํด๋ž˜์Šค์—์„œ ํ‰๊ท  AP ๊ฐ’์„ ๊ณ„์‚ฐํ•˜์—ฌ AP์˜ ๊ฐœ๋…์„ ํ™•์žฅํ•ฉ๋‹ˆ๋‹ค. ์ด๋Š” ๋‹ค์ค‘ ํด๋ž˜์Šค ๊ฐ์ฒด ๊ฐ์ง€ ์‹œ๋‚˜๋ฆฌ์˜ค์—์„œ ๋ชจ๋ธ ์„ฑ๋Šฅ์— ๋Œ€ํ•œ ์ข…ํ•ฉ์ ์ธ ํ‰๊ฐ€๋ฅผ ์ œ๊ณตํ•˜๋Š” ๋ฐ ์œ ์šฉํ•ฉ๋‹ˆ๋‹ค.

  • ์ •ํ™•๋„ ๋ฐ ํšŒ์ˆ˜์œจ: ์ •ํ™•๋„๋Š” ๋ชจ๋“  ์–‘์„ฑ ์˜ˆ์ธก ์ค‘ ์ •์–‘์„ฑ ๋น„์œจ์„ ์ •๋Ÿ‰ํ™”ํ•˜์—ฌ ์˜คํƒ์„ ํ”ผํ•˜๋Š” ๋ชจ๋ธ์˜ ๋Šฅ๋ ฅ์„ ํ‰๊ฐ€ํ•ฉ๋‹ˆ๋‹ค. ๋ฐ˜๋ฉด์— Recall์€ ๋ชจ๋“  ์‹ค์ œ ์–‘์„ฑ ์ค‘ ์ •ํƒ์˜ ๋น„์œจ์„ ๊ณ„์‚ฐํ•˜์—ฌ ํด๋ž˜์Šค์˜ ๋ชจ๋“  ์ธ์Šคํ„ด์Šค๋ฅผ ํƒ์ง€ํ•˜๋Š” ๋ชจ๋ธ์˜ ๋Šฅ๋ ฅ์„ ์ธก์ •ํ•ฉ๋‹ˆ๋‹ค.

  • F1 ์ ์ˆ˜: F1 ์ ์ˆ˜๋Š” ์ •ํ™•๋„์™€ ํšŒ์ˆ˜์œจ์˜ ์กฐํ™” ํ‰๊ท ์œผ๋กœ, ์˜คํƒ๊ณผ ์˜คํƒ์„ ๋ชจ๋‘ ๊ณ ๋ คํ•˜๋ฉด์„œ ๋ชจ๋ธ์˜ ์„ฑ๋Šฅ์„ ๊ท ํ˜• ์žˆ๊ฒŒ ํ‰๊ฐ€ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

YOLOv8 ๋ชจ๋ธ์— ๋Œ€ํ•œ ์ง€ํ‘œ ๊ณ„์‚ฐ ๋ฐฉ๋ฒ•

์ด์ œ ์œ„์—์„œ ์„ค๋ช…ํ•œ ํ‰๊ฐ€ ์ง€ํ‘œ๋ฅผ ๊ณ„์‚ฐํ•˜๋Š” ๋ฐ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ๋Š” YOLOv8 ์˜ ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ๋ชจ๋“œ๋ฅผ ์‚ดํŽด๋ณผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ๋ชจ๋“œ ์‚ฌ์šฉ์€ ๊ฐ„๋‹จํ•ฉ๋‹ˆ๋‹ค. ํ•™์Šต๋œ ๋ชจ๋ธ์ด ์žˆ์œผ๋ฉด model.val() ํ•จ์ˆ˜๋ฅผ ํ˜ธ์ถœํ•˜๋ฉด ๋ฉ๋‹ˆ๋‹ค. ๊ทธ๋Ÿฌ๋ฉด ์ด ํ•จ์ˆ˜๋Š” ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ๋ฐ์ดํ„ฐ ์„ธํŠธ๋ฅผ ์ฒ˜๋ฆฌํ•˜๊ณ  ๋‹ค์–‘ํ•œ ์„ฑ๋Šฅ ์ง€ํ‘œ๋ฅผ ๋ฐ˜ํ™˜ํ•ฉ๋‹ˆ๋‹ค. ํ•˜์ง€๋งŒ ์ด๋Ÿฌํ•œ ๋ฉ”ํŠธ๋ฆญ์€ ๋ฌด์—‡์„ ์˜๋ฏธํ• ๊นŒ์š”? ๊ทธ๋ฆฌ๊ณ  ์ด๋ฅผ ์–ด๋–ป๊ฒŒ ํ•ด์„ํ•ด์•ผ ํ• ๊นŒ์š”?

์ถœ๋ ฅ ํ•ด์„ํ•˜๊ธฐ

model.val() ํ•จ์ˆ˜์˜ ์ถœ๋ ฅ์„ ์„ธ๋ถ„ํ™”ํ•˜์—ฌ ์ถœ๋ ฅ์˜ ๊ฐ ์„ธ๊ทธ๋จผํŠธ๋ฅผ ์ดํ•ดํ•ด ๋ณด๊ฒ ์Šต๋‹ˆ๋‹ค.

ํด๋ž˜์Šค๋ณ„ ๋ฉ”ํŠธ๋ฆญ

์ถœ๋ ฅ์˜ ์„น์…˜ ์ค‘ ํ•˜๋‚˜๋Š” ์„ฑ๋Šฅ ๋ฉ”ํŠธ๋ฆญ์˜ ํด๋ž˜์Šค๋ณ„ ๋ถ„์„์ž…๋‹ˆ๋‹ค. ์ด ์„ธ๋ถ„ํ™”๋œ ์ •๋ณด๋Š” ํŠนํžˆ ๋‹ค์–‘ํ•œ ๋ฒ”์œ„์˜ ๊ฐœ์ฒด ๋ฒ”์ฃผ๊ฐ€ ์žˆ๋Š” ๋ฐ์ดํ„ฐ ์„ธํŠธ์—์„œ ๊ฐ ํŠน์ • ํด๋ž˜์Šค์— ๋Œ€ํ•ด ๋ชจ๋ธ์ด ์–ผ๋งˆ๋‚˜ ์ž˜ ์ž‘๋™ํ•˜๋Š”์ง€ ์ดํ•ดํ•˜๋ ค๊ณ  ํ•  ๋•Œ ์œ ์šฉํ•ฉ๋‹ˆ๋‹ค. ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ๊ฐ ํด๋ž˜์Šค์— ๋Œ€ํ•ด ๋‹ค์Œ์ด ์ œ๊ณต๋ฉ๋‹ˆ๋‹ค:

  • ํด๋ž˜์Šค: '์‚ฌ๋žŒ', '์ž๋™์ฐจ', '๊ฐœ' ๋“ฑ ๊ฐœ์ฒด ํด๋ž˜์Šค์˜ ์ด๋ฆ„์„ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค.

  • ์ด๋ฏธ์ง€: ์ด ๋ฉ”ํŠธ๋ฆญ์€ ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ์„ธํŠธ์—์„œ ๊ฐœ์ฒด ํด๋ž˜์Šค๋ฅผ ํฌํ•จํ•˜๋Š” ์ด๋ฏธ์ง€์˜ ์ˆ˜๋ฅผ ์•Œ๋ ค์ค๋‹ˆ๋‹ค.

  • ์ธ์Šคํ„ด์Šค: ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ์„ธํŠธ์˜ ๋ชจ๋“  ์ด๋ฏธ์ง€์—์„œ ํด๋ž˜์Šค๊ฐ€ ํ‘œ์‹œ๋˜๋Š” ํšŸ์ˆ˜๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

  • Box(P, R, mAP50, mAP50-95): ์ด ๋ฉ”ํŠธ๋ฆญ์€ ๋ฌผ์ฒด๋ฅผ ๊ฐ์ง€ํ•˜๋Š” ๋ชจ๋ธ์˜ ์„ฑ๋Šฅ์— ๋Œ€ํ•œ ์ธ์‚ฌ์ดํŠธ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค:

    • P(์ •ํ™•๋„): ๊ฐ์ง€๋œ ๊ฐœ์ฒด์˜ ์ •ํ™•๋„๋กœ, ์–ผ๋งˆ๋‚˜ ๋งŽ์€ ๊ฐœ์ฒด๋ฅผ ์ •ํ™•ํ•˜๊ฒŒ ๊ฐ์ง€ํ–ˆ๋Š”์ง€๋ฅผ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค.

    • R(๋ฆฌ์ฝœ): ์ด๋ฏธ์ง€์—์„œ ๊ฐ์ฒด์˜ ๋ชจ๋“  ์ธ์Šคํ„ด์Šค๋ฅผ ์‹๋ณ„ํ•˜๋Š” ๋ชจ๋ธ์˜ ๊ธฐ๋Šฅ์ž…๋‹ˆ๋‹ค.

    • mAP50: ํ‰๊ท  ํ‰๊ท  ์ •๋ฐ€๋„(IoU) ์ž„๊ณ„๊ฐ’ 0.50์—์„œ ๊ณ„์‚ฐ๋œ ํ‰๊ท  ์ •๋ฐ€๋„์ž…๋‹ˆ๋‹ค. '์‰ฌ์šด' ํƒ์ง€๋งŒ ๊ณ ๋ คํ•œ ๋ชจ๋ธ์˜ ์ •ํ™•๋„๋ฅผ ์ธก์ •ํ•œ ๊ฐ’์ž…๋‹ˆ๋‹ค.

    • mAP50-95: 0.50~0.95 ๋ฒ”์œ„์˜ ๋‹ค์–‘ํ•œ IoU ์ž„๊ณ„๊ฐ’์—์„œ ๊ณ„์‚ฐ๋œ ํ‰๊ท  ํ‰๊ท  ์ •๋ฐ€๋„์˜ ํ‰๊ท ์ž…๋‹ˆ๋‹ค. ๋‹ค์–‘ํ•œ ํƒ์ง€ ๋‚œ์ด๋„ ์ˆ˜์ค€์—์„œ ๋ชจ๋ธ์˜ ์„ฑ๋Šฅ์„ ์ข…ํ•ฉ์ ์œผ๋กœ ํŒŒ์•…ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์†๋„ ๋ฉ”ํŠธ๋ฆญ

์ถ”๋ก  ์†๋„๋Š” ํŠนํžˆ ์‹ค์‹œ๊ฐ„ ๊ฐ์ฒด ๊ฐ์ง€ ์‹œ๋‚˜๋ฆฌ์˜ค์—์„œ ์ •ํ™•๋„๋งŒํผ์ด๋‚˜ ์ค‘์š”ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์ด ์„น์…˜์—์„œ๋Š” ์ „์ฒ˜๋ฆฌ๋ถ€ํ„ฐ ํ›„์ฒ˜๋ฆฌ๊นŒ์ง€ ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ํ”„๋กœ์„ธ์Šค์˜ ๋‹ค์–‘ํ•œ ๋‹จ๊ณ„์— ๊ฑธ๋ฆฌ๋Š” ์‹œ๊ฐ„์„ ์„ธ๋ถ„ํ™”ํ•˜์—ฌ ์„ค๋ช…ํ•ฉ๋‹ˆ๋‹ค.

COCO ์ง€ํ‘œ ํ‰๊ฐ€

COCO ๋ฐ์ดํ„ฐ ์„ธํŠธ์—์„œ ์œ ํšจ์„ฑ์„ ๊ฒ€์‚ฌํ•˜๋Š” ์‚ฌ์šฉ์ž์˜ ๊ฒฝ์šฐ, COCO ํ‰๊ฐ€ ์Šคํฌ๋ฆฝํŠธ๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ์ถ”๊ฐ€ ์ง€ํ‘œ๊ฐ€ ๊ณ„์‚ฐ๋ฉ๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ๋ฉ”ํŠธ๋ฆญ์€ ๋‹ค์–‘ํ•œ IoU ์ž„๊ณ„๊ฐ’๊ณผ ๋‹ค์–‘ํ•œ ํฌ๊ธฐ์˜ ์˜ค๋ธŒ์ ํŠธ์— ๋Œ€ํ•œ ์ •ํ™•๋„ ๋ฐ ๋ฆฌ์ฝœ์— ๋Œ€ํ•œ ์ธ์‚ฌ์ดํŠธ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

์‹œ๊ฐ์  ์ถœ๋ ฅ

model.val() ํ•จ์ˆ˜๋Š” ์ˆซ์ž ๋ฉ”ํŠธ๋ฆญ์„ ์ƒ์„ฑํ•˜๋Š” ๊ฒƒ ์™ธ์—๋„ ๋ชจ๋ธ์˜ ์„ฑ๋Šฅ์„ ๋ณด๋‹ค ์ง๊ด€์ ์œผ๋กœ ์ดํ•ดํ•  ์ˆ˜ ์žˆ๋Š” ์‹œ๊ฐ์  ์ถœ๋ ฅ์„ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค. ๋‹ค์Œ์€ ์˜ˆ์ƒํ•  ์ˆ˜ ์žˆ๋Š” ์‹œ๊ฐ์  ์ถœ๋ ฅ์— ๋Œ€ํ•œ ๋ถ„์„์ž…๋‹ˆ๋‹ค:

  • F1 ์ ์ˆ˜ ์ปค๋ธŒ(F1_curve.png): ์ด ๊ณก์„ ์€ ๋‹ค์–‘ํ•œ ์ž„๊ณ„๊ฐ’์— ๊ฑธ์นœ F1 ์ ์ˆ˜๋ฅผ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค. ์ด ๊ณก์„ ์„ ํ•ด์„ํ•˜๋ฉด ๋‹ค์–‘ํ•œ ์ž„๊ณ„๊ฐ’์—์„œ ์˜คํƒ๊ณผ ์˜คํƒ ์‚ฌ์ด์˜ ๋ชจ๋ธ ๊ท ํ˜•์— ๋Œ€ํ•œ ์ธ์‚ฌ์ดํŠธ๋ฅผ ์–ป์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ์ •๋ฐ€๋„-๋ฆฌ์ฝœ ์ปค๋ธŒ(PR_curve.png): ๋ชจ๋“  ๋ถ„๋ฅ˜ ๋ฌธ์ œ์— ๋Œ€ํ•œ ํ†ตํ•ฉ ์‹œ๊ฐํ™”์ธ ์ด ๊ณก์„ ์€ ๋‹ค์–‘ํ•œ ์ž„๊ณ„๊ฐ’์—์„œ ์ •ํ™•๋„์™€ ํšŒ์ƒ๋ฅ  ์‚ฌ์ด์˜ ์ ˆ์ถฉ์ ์„ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค. ๋ถˆ๊ท ํ˜•ํ•œ ํด๋ž˜์Šค๋ฅผ ๋‹ค๋ฃฐ ๋•Œ ํŠนํžˆ ์ค‘์š”ํ•ด์ง‘๋‹ˆ๋‹ค.

  • ์ •๋ฐ€ ์ปค๋ธŒ(P_curve.png): ๋‹ค์–‘ํ•œ ์ž„๊ณ„๊ฐ’์—์„œ์˜ ์ •๋ฐ€๋„ ๊ฐ’์„ ๊ทธ๋ž˜ํ”ฝ์œผ๋กœ ํ‘œํ˜„ํ•œ ๊ฒƒ์ž…๋‹ˆ๋‹ค. ์ด ๊ณก์„ ์€ ์ž„๊ณ„๊ฐ’์ด ๋ณ€ํ•จ์— ๋”ฐ๋ผ ์ •๋ฐ€๋„๊ฐ€ ์–ด๋–ป๊ฒŒ ๋ณ€ํ•˜๋Š”์ง€๋ฅผ ์ดํ•ดํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋ฉ๋‹ˆ๋‹ค.

  • ๋ฆฌ์ฝœ ์ปค๋ธŒ(R_curve.png): ์ด์— ๋”ฐ๋ผ ์ด ๊ทธ๋ž˜ํ”„๋Š” ๋‹ค์–‘ํ•œ ์ž„๊ณ„๊ฐ’์— ๋”ฐ๋ผ ๋ฆฌ์ฝœ ๊ฐ’์ด ์–ด๋–ป๊ฒŒ ๋ณ€ํ•˜๋Š”์ง€๋ฅผ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค.

  • ํ˜ผ๋™ ๋งคํŠธ๋ฆญ์Šค (confusion_matrix.png): ํ˜ผ๋™ ํ–‰๋ ฌ์€ ๊ฐ ํด๋ž˜์Šค์— ๋Œ€ํ•œ ์ง„์–‘์„ฑ, ์ง„์Œ์„ฑ, ์˜คํƒ, ์˜คํƒ์˜ ๊ฐœ์ˆ˜๋ฅผ ๋ณด์—ฌ์ฃผ๋Š” ๊ฒฐ๊ณผ์— ๋Œ€ํ•œ ์ž์„ธํ•œ ๋ณด๊ธฐ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

  • ์ •๊ทœํ™”๋œ ํ˜ผ๋™ ํ–‰๋ ฌ(confusion_matrix_normalized.png): ์ด ์‹œ๊ฐํ™”๋Š” ํ˜ผ๋™ ํ–‰๋ ฌ์˜ ์ •๊ทœํ™”๋œ ๋ฒ„์ „์ž…๋‹ˆ๋‹ค. ์›์‹œ ๊ฐœ์ˆ˜๊ฐ€ ์•„๋‹Œ ๋น„์œจ๋กœ ๋ฐ์ดํ„ฐ๋ฅผ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค. ์ด ํ˜•์‹์„ ์‚ฌ์šฉํ•˜๋ฉด ํด๋ž˜์Šค ๊ฐ„ ์„ฑ๋Šฅ์„ ๋” ๊ฐ„๋‹จํ•˜๊ฒŒ ๋น„๊ตํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ๋ฐฐ์น˜ ๋ ˆ์ด๋ธ”(val_batchX_labels.jpg): ์ด ์ด๋ฏธ์ง€๋Š” ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ๋ฐ์ดํ„ฐ ์„ธํŠธ์—์„œ ๋ณ„๊ฐœ์˜ ๋ฐฐ์น˜์— ๋Œ€ํ•œ ๊ธฐ์ค€๊ฐ’ ๋ ˆ์ด๋ธ”์„ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค. ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ๋”ฐ๋ผ ๊ฐœ์ฒด์˜ ์ข…๋ฅ˜์™€ ๊ฐ ์œ„์น˜๋ฅผ ๋ช…ํ™•ํ•˜๊ฒŒ ํŒŒ์•…ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ๋ฐฐ์น˜ ์˜ˆ์ธก(val_batchX_pred.jpg): ๋ผ๋ฒจ ์ด๋ฏธ์ง€์™€ ๋Œ€์กฐ๋˜๋Š” ์ด ์‹œ๊ฐ ์ž๋ฃŒ๋Š” YOLOv8 ๋ชจ๋ธ์ด ๊ฐ ๋ฐฐ์น˜์— ๋Œ€ํ•ด ์˜ˆ์ธกํ•œ ๊ฒฐ๊ณผ๋ฅผ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค. ์ด๋ฅผ ๋ ˆ์ด๋ธ” ์ด๋ฏธ์ง€์™€ ๋น„๊ตํ•˜๋ฉด ๋ชจ๋ธ์ด ์‹œ๊ฐ์ ์œผ๋กœ ๊ฐœ์ฒด๋ฅผ ์–ผ๋งˆ๋‚˜ ์ž˜ ๊ฐ์ง€ํ•˜๊ณ  ๋ถ„๋ฅ˜ํ•˜๋Š”์ง€ ์‰ฝ๊ฒŒ ํ‰๊ฐ€ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

๊ฒฐ๊ณผ ์ €์žฅ

๋‚˜์ค‘์— ์ฐธ์กฐํ•  ์ˆ˜ ์žˆ๋„๋ก ๊ฒฐ๊ณผ๋Š” ์ผ๋ฐ˜์ ์œผ๋กœ runs/detect/val์ด๋ผ๋Š” ์ด๋ฆ„์˜ ๋””๋ ‰ํ„ฐ๋ฆฌ์— ์ €์žฅ๋ฉ๋‹ˆ๋‹ค.

์˜ฌ๋ฐ”๋ฅธ ์ง€ํ‘œ ์„ ํƒ

ํ‰๊ฐ€ํ•  ์˜ฌ๋ฐ”๋ฅธ ์ง€ํ‘œ๋ฅผ ์„ ํƒํ•˜๋Š” ๊ฒƒ์€ ์ข…์ข… ํŠน์ • ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜์— ๋”ฐ๋ผ ๋‹ฌ๋ผ์ง‘๋‹ˆ๋‹ค.

  • mAP: ๋ชจ๋ธ ์„ฑ๋Šฅ์— ๋Œ€ํ•œ ๊ด‘๋ฒ”์œ„ํ•œ ํ‰๊ฐ€์— ์ ํ•ฉํ•ฉ๋‹ˆ๋‹ค.

  • IoU: ์ •ํ™•ํ•œ ๊ฐ์ฒด ์œ„์น˜๊ฐ€ ์ค‘์š”ํ•œ ๊ฒฝ์šฐ ํ•„์ˆ˜์ ์ž…๋‹ˆ๋‹ค.

  • ์ •ํ™•์„ฑ: ์˜คํƒ์ง€๋ฅผ ์ตœ์†Œํ™”ํ•˜๋Š” ๊ฒƒ์ด ์ตœ์šฐ์„  ๊ณผ์ œ์ผ ๋•Œ ์ค‘์š”ํ•ฉ๋‹ˆ๋‹ค.

  • ๋ฆฌ์ฝœ: ๊ฐ์ฒด์˜ ๋ชจ๋“  ์ธ์Šคํ„ด์Šค๋ฅผ ๊ฐ์ง€ํ•˜๋Š” ๊ฒƒ์ด ์ค‘์š”ํ•œ ๊ฒฝ์šฐ ํ•„์ˆ˜์ ์ž…๋‹ˆ๋‹ค.

  • F1 ์ ์ˆ˜: ์ •ํ™•๋„์™€ ํšŒ์ƒ๋ฅ  ๊ฐ„์˜ ๊ท ํ˜•์ด ํ•„์š”ํ•  ๋•Œ ์œ ์šฉํ•ฉ๋‹ˆ๋‹ค.

์‹ค์‹œ๊ฐ„ ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜์˜ ๊ฒฝ์šฐ, ์ ์‹œ์— ๊ฒฐ๊ณผ๋ฅผ ์–ป๊ธฐ ์œ„ํ•ด์„œ๋Š” FPS(์ดˆ๋‹น ํ”„๋ ˆ์ž„ ์ˆ˜) ๋ฐ ์ง€์—ฐ ์‹œ๊ฐ„๊ณผ ๊ฐ™์€ ์†๋„ ์ง€ํ‘œ๊ฐ€ ๋งค์šฐ ์ค‘์š”ํ•ฉ๋‹ˆ๋‹ค.

๊ฒฐ๊ณผ ํ•ด์„

๋ฉ”ํŠธ๋ฆญ์„ ์ดํ•ดํ•˜๋Š” ๊ฒƒ์ด ์ค‘์š”ํ•ฉ๋‹ˆ๋‹ค. ๋‹ค์Œ์€ ์ผ๋ฐ˜์ ์œผ๋กœ ๊ด€์ฐฐ๋˜๋Š” ๋ช‡ ๊ฐ€์ง€ ๋‚ฎ์€ ์ ์ˆ˜๊ฐ€ ์‹œ์‚ฌํ•˜๋Š” ๋ฐ”์ž…๋‹ˆ๋‹ค:

  • ๋‚ฎ์€ ๋งต: ๋ชจ๋ธ์— ์ผ๋ฐ˜์ ์ธ ๊ฐœ์„ ์ด ํ•„์š”ํ•  ์ˆ˜ ์žˆ์Œ์„ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค.

  • ๋‚ฎ์€ IoU: ๋ชจ๋ธ์ด ์˜ค๋ธŒ์ ํŠธ๋ฅผ ์ •ํ™•ํ•˜๊ฒŒ ์ฐพ์•„๋‚ด๋Š” ๋ฐ ์–ด๋ ค์›€์„ ๊ฒช๊ณ  ์žˆ์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋‹ค๋ฅธ ๋ฐ”์šด๋”ฉ ๋ฐ•์Šค ๋ฉ”์„œ๋“œ๊ฐ€ ๋„์›€์ด ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ๋‚ฎ์€ ์ •๋ฐ€๋„: ๋ชจ๋ธ์ด ์กด์žฌํ•˜์ง€ ์•Š๋Š” ๊ฐœ์ฒด๋ฅผ ๋„ˆ๋ฌด ๋งŽ์ด ๊ฐ์ง€ํ•˜๊ณ  ์žˆ์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์‹ ๋ขฐ๋„ ์ž„๊ณ„๊ฐ’์„ ์กฐ์ •ํ•˜๋ฉด ์ด ๋ฌธ์ œ๋ฅผ ์ค„์ผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ๋‚ฎ์€ ๋ฆฌ์ฝœ: ๋ชจ๋ธ์— ์‹ค์ œ ๊ฐœ์ฒด๊ฐ€ ๋ˆ„๋ฝ๋˜์—ˆ์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ํŠน์ง• ์ถ”์ถœ์„ ๊ฐœ์„ ํ•˜๊ฑฐ๋‚˜ ๋” ๋งŽ์€ ๋ฐ์ดํ„ฐ๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ๋„์›€์ด ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ๋ถˆ๊ท ํ˜•ํ•œ F1 ์ ์ˆ˜: ์ •ํ™•๋„์™€ ํšŒ์ƒ๋„ ์‚ฌ์ด์— ๋ถˆ๊ท ํ˜•์ด ์žˆ์Šต๋‹ˆ๋‹ค.

  • ์ˆ˜์—…๋ณ„ AP: ์—ฌ๊ธฐ์„œ ๋‚ฎ์€ ์ ์ˆ˜๋Š” ๋ชจ๋ธ์ด ์–ด๋ ค์›€์„ ๊ฒช๊ณ  ์žˆ๋Š” ์ˆ˜์—…์„ ๊ฐ•์กฐํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์‚ฌ๋ก€ ์—ฐ๊ตฌ

์‹ค์ œ ์‚ฌ๋ก€๋ฅผ ํ†ตํ•ด ์ด๋Ÿฌํ•œ ๋ฉ”ํŠธ๋ฆญ์ด ์‹ค์ œ๋กœ ์–ด๋–ป๊ฒŒ ์ž‘๋™ํ•˜๋Š”์ง€ ๋ช…ํ™•ํžˆ ์•Œ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์‚ฌ๋ก€ 1

  • ์ƒํ™ฉ: mAP์™€ F1 ์ ์ˆ˜๋Š” ์ตœ์ ์ด ์•„๋‹ˆ์ง€๋งŒ, ๋ฆฌ์ฝœ์€ ์ข‹์ง€๋งŒ ์ •ํ™•๋„๋Š” ๊ทธ๋ ‡์ง€ ์•Š์Šต๋‹ˆ๋‹ค.

  • ํ•ด์„ ๋ฐ ์กฐ์น˜: ์ž˜๋ชป๋œ ๊ฐ์ง€๊ฐ€ ๋„ˆ๋ฌด ๋งŽ์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์‹ ๋ขฐ๋„ ์ž„๊ณ„๊ฐ’์„ ๊ฐ•ํ™”ํ•˜๋ฉด ์ด๋Ÿฌํ•œ ์˜ค๋ฅ˜๋ฅผ ์ค„์ผ ์ˆ˜ ์žˆ์ง€๋งŒ ํšŒ์ƒ๋ฅ ์ด ์•ฝ๊ฐ„ ๊ฐ์†Œํ•  ์ˆ˜๋„ ์žˆ์Šต๋‹ˆ๋‹ค.

์‚ฌ๋ก€ 2

  • ์ƒํ™ฉ: mAP ๋ฐ ๋ฆฌ์ฝœ์€ ํ—ˆ์šฉ๋˜์ง€๋งŒ IoU๋Š” ๋ถ€์กฑํ•ฉ๋‹ˆ๋‹ค.

  • ํ•ด์„ ๋ฐ ์กฐ์น˜: ๋ชจ๋ธ์ด ๊ฐ์ฒด๋ฅผ ์ž˜ ๊ฐ์ง€ํ•˜์ง€๋งŒ ์ •ํ™•ํ•˜๊ฒŒ ์œ„์น˜๋ฅผ ํŒŒ์•…ํ•˜์ง€ ๋ชปํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋ฐ”์šด๋”ฉ ๋ฐ•์Šค ์˜ˆ์ธก์„ ์„ธ๋ถ„ํ™”ํ•˜๋ฉด ๋„์›€์ด ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์‚ฌ๋ก€ 3

  • ์ƒํ™ฉ: ์ผ๋ถ€ ํด๋ž˜์Šค๋Š” ์ „์ฒด ๋งต์ด ๊ดœ์ฐฎ๋”๋ผ๋„ ๋‹ค๋ฅธ ํด๋ž˜์Šค๋ณด๋‹ค AP๊ฐ€ ํ›จ์”ฌ ๋‚ฎ์Šต๋‹ˆ๋‹ค.

  • ํ•ด์„ ๋ฐ ํ–‰๋™: ์ด๋Ÿฌํ•œ ํด๋ž˜์Šค๋Š” ๋ชจ๋ธ์— ๋” ๊นŒ๋‹ค๋กœ์šธ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ํด๋ž˜์Šค์— ๋” ๋งŽ์€ ๋ฐ์ดํ„ฐ๋ฅผ ์‚ฌ์šฉํ•˜๊ฑฐ๋‚˜ ํ›ˆ๋ จ ์ค‘์— ํด๋ž˜์Šค ๊ฐ€์ค‘์น˜๋ฅผ ์กฐ์ •ํ•˜๋ฉด ๋„์›€์ด ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์—ฐ๊ฒฐ ๋ฐ ๊ณต๋™ ์ž‘์—…

์• ํ˜ธ๊ฐ€ ๋ฐ ์ „๋ฌธ๊ฐ€๋กœ ๊ตฌ์„ฑ๋œ ์ปค๋ฎค๋‹ˆํ‹ฐ๋ฅผ ํ™œ์šฉํ•˜๋ฉด YOLOv8 ์„ ํ†ตํ•ด ํ•™์Šต, ๋ฌธ์ œ ํ•ด๊ฒฐ ๋ฐ ๋„คํŠธ์›Œํ‚น์„ ์ด‰์ง„ํ•  ์ˆ˜ ์žˆ๋Š” ๋ช‡ ๊ฐ€์ง€ ๋ฐฉ๋ฒ•์„ ์†Œ๊ฐœํ•ฉ๋‹ˆ๋‹ค.

๋” ๋„“์€ ์ปค๋ฎค๋‹ˆํ‹ฐ์™€ ์†Œํ†ตํ•˜๊ธฐ

  • GitHub ์ด์Šˆ: YOLOv8 ๋ฆฌํฌ์ง€ํ† ๋ฆฌ์—๋Š” ์งˆ๋ฌธํ•˜๊ณ , ๋ฒ„๊ทธ๋ฅผ ์‹ ๊ณ ํ•˜๊ณ , ์ƒˆ๋กœ์šด ๊ธฐ๋Šฅ์„ ์ œ์•ˆํ•  ์ˆ˜ ์žˆ๋Š” ์ด์Šˆ ํƒญ์ด ์žˆ์Šต๋‹ˆ๋‹ค. ์ปค๋ฎค๋‹ˆํ‹ฐ์™€ ๊ด€๋ฆฌ์ž๊ฐ€ ํ™œ๋ฐœํ•˜๊ฒŒ ํ™œ๋™ํ•˜๊ณ  ์žˆ์œผ๋ฉฐ ํŠน์ • ๋ฌธ์ œ์— ๋Œ€ํ•œ ๋„์›€์„ ๋ฐ›์„ ์ˆ˜ ์žˆ๋Š” ์ข‹์€ ๊ณณ์ž…๋‹ˆ๋‹ค.

  • Ultralytics ๋””์Šค์ฝ”๋“œ ์„œ๋ฒ„: Ultralytics ๋‹ค๋ฅธ ์‚ฌ์šฉ์ž ๋ฐ ๊ฐœ๋ฐœ์ž์™€ ์†Œํ†ตํ•  ์ˆ˜ ์žˆ๋Š” ๋””์Šค์ฝ”๋“œ ์„œ๋ฒ„๊ฐ€ ์žˆ์Šต๋‹ˆ๋‹ค.

๊ณต์‹ ๋ฌธ์„œ ๋ฐ ๋ฆฌ์†Œ์Šค:

  • Ultralytics YOLOv8 ๋ฌธ์„œ: ๊ณต์‹ ๋ฌธ์„œ์—์„œ๋Š” ์„ค์น˜, ์‚ฌ์šฉ ๋ฐ ๋ฌธ์ œ ํ•ด๊ฒฐ์— ๋Œ€ํ•œ ๊ฐ€์ด๋“œ์™€ ํ•จ๊ป˜ YOLOv8 ์— ๋Œ€ํ•œ ํฌ๊ด„์ ์ธ ๊ฐœ์š”๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

์ด๋Ÿฌํ•œ ๋ฆฌ์†Œ์Šค๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋  ๋ฟ๋งŒ ์•„๋‹ˆ๋ผ YOLOv8 ์ปค๋ฎค๋‹ˆํ‹ฐ์—์„œ ์ตœ์‹  ํŠธ๋ Œ๋“œ์™€ ๋ชจ๋ฒ” ์‚ฌ๋ก€๋ฅผ ๊ณ„์† ์—…๋ฐ์ดํŠธํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

๊ฒฐ๋ก 

์ด ๊ฐ€์ด๋“œ์—์„œ๋Š” YOLOv8 ์— ๋Œ€ํ•œ ํ•„์ˆ˜ ์„ฑ๋Šฅ ์ง€ํ‘œ๋ฅผ ์ž์„ธํžˆ ์‚ดํŽด๋ณด์•˜์Šต๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ์ง€ํ‘œ๋Š” ๋ชจ๋ธ์˜ ์„ฑ๋Šฅ์„ ์ดํ•ดํ•˜๋Š” ๋ฐ ํ•ต์‹ฌ์ ์ธ ์—ญํ• ์„ ํ•˜๋ฉฐ ๋ชจ๋ธ์„ ๋ฏธ์„ธ ์กฐ์ •ํ•˜๋ ค๋Š” ๋ชจ๋“  ์‚ฌ๋žŒ์—๊ฒŒ ํ•„์ˆ˜์ ์ž…๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ์ง€ํ‘œ๋Š” ๋ชจ๋ธ์„ ๊ฐœ์„ ํ•˜๊ณ  ์‹ค์ œ ์ƒํ™ฉ์—์„œ ๋ชจ๋ธ์ด ํšจ๊ณผ์ ์œผ๋กœ ์ž‘๋™ํ•˜๋Š”์ง€ ํ™•์ธํ•˜๋Š” ๋ฐ ํ•„์š”ํ•œ ์ธ์‚ฌ์ดํŠธ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

YOLOv8 ๋ฐ Ultralytics ์ปค๋ฎค๋‹ˆํ‹ฐ๋Š” ๊ท€์ค‘ํ•œ ์ž์‚ฐ์ž„์„ ๊ธฐ์–ตํ•˜์„ธ์š”. ๋™๋ฃŒ ๊ฐœ๋ฐœ์ž ๋ฐ ์ „๋ฌธ๊ฐ€์™€ ์†Œํ†ตํ•˜๋ฉด ํ‘œ์ค€ ๋ฌธ์„œ์—์„œ ์ฐพ์„ ์ˆ˜ ์—†๋Š” ์ธ์‚ฌ์ดํŠธ์™€ ์†”๋ฃจ์…˜์„ ์–ป์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๊ฐ์ฒด ํƒ์ง€๋ฅผ ์ง„ํ–‰ํ•˜๋ฉด์„œ ๋ฐฐ์›€์˜ ์ •์‹ ์„ ์žƒ์ง€ ๋ง๊ณ  ์ƒˆ๋กœ์šด ์ „๋žต์„ ์‹คํ—˜ํ•˜๊ณ  ๊ทธ ๊ฒฐ๊ณผ๋ฅผ ๊ณต์œ ํ•˜์„ธ์š”. ๊ทธ๋ ‡๊ฒŒ ํ•จ์œผ๋กœ์จ ์ปค๋ฎค๋‹ˆํ‹ฐ์˜ ์ง‘๋‹จ์  ์ง€ํ˜œ์— ๊ธฐ์—ฌํ•˜๊ณ  ์ปค๋ฎค๋‹ˆํ‹ฐ์˜ ์„ฑ์žฅ์„ ๋ณด์žฅํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

ํ–‰๋ณตํ•œ ๋ฌผ์ฒด ๊ฐ์ง€!



์ƒ์„ฑ 2023-11-12, ์—…๋ฐ์ดํŠธ 2024-03-02
์ž‘์„ฑ์ž: glenn-jocher (4), RizwanMunawar (1), abirami-vina (1)

๋Œ“๊ธ€