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

๋จธ์‹  ๋Ÿฌ๋‹ ๋ชจ๋ฒ” ์‚ฌ๋ก€ ๋ฐ ๋ชจ๋ธ ํ•™์Šต์„ ์œ„ํ•œ ํŒ

์†Œ๊ฐœ

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

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

๋จธ์‹ ๋Ÿฌ๋‹ ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๋Š” ๋ฐฉ๋ฒ•

์ปดํ“จํ„ฐ ๋น„์ „ ๋ชจ๋ธ์€ ์˜ค๋ฅ˜๋ฅผ ์ตœ์†Œํ™”ํ•˜๊ธฐ ์œ„ํ•ด ๋‚ด๋ถ€ ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์กฐ์ •ํ•˜์—ฌ ํ•™์Šตํ•ฉ๋‹ˆ๋‹ค. ์ฒ˜์Œ์—๋Š” ๋ชจ๋ธ์— ๋ผ๋ฒจ์ด ์ง€์ •๋œ ๋Œ€๊ทœ๋ชจ ์ด๋ฏธ์ง€ ์„ธํŠธ๊ฐ€ ์ œ๊ณต๋ฉ๋‹ˆ๋‹ค. ์ด ์ด๋ฏธ์ง€์— ๋ฌด์—‡์ด ์žˆ๋Š”์ง€ ์˜ˆ์ธกํ•˜๊ณ  ์˜ˆ์ธก์„ ์‹ค์ œ ๋ผ๋ฒจ์ด๋‚˜ ๋‚ด์šฉ๊ณผ ๋น„๊ตํ•˜์—ฌ ์˜ค์ฐจ๋ฅผ ๊ณ„์‚ฐํ•ฉ๋‹ˆ๋‹ค. ์ด๋Ÿฌํ•œ ์˜ค์ฐจ๋Š” ๋ชจ๋ธ์˜ ์˜ˆ์ธก์ด ์‹ค์ œ ๊ฐ’๊ณผ ์–ผ๋งˆ๋‚˜ ์ฐจ์ด๊ฐ€ ๋‚˜๋Š”์ง€๋ฅผ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค.

ํ•™์Šตํ•˜๋Š” ๋™์•ˆ ๋ชจ๋ธ์€ ๋ฐ˜๋ณต์ ์œผ๋กœ ์˜ˆ์ธก์„ ํ•˜๊ณ , ์˜ค๋ฅ˜๋ฅผ ๊ณ„์‚ฐํ•˜๊ณ , ์—ญ์ „ํŒŒ๋ผ๋Š” ํ”„๋กœ์„ธ์Šค๋ฅผ ํ†ตํ•ด ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์—…๋ฐ์ดํŠธํ•ฉ๋‹ˆ๋‹ค. ์ด ๊ณผ์ •์—์„œ ๋ชจ๋ธ์€ ๋‚ด๋ถ€ ํŒŒ๋ผ๋ฏธํ„ฐ(weights and biases)๋ฅผ ์กฐ์ •ํ•˜์—ฌ ์˜ค๋ฅ˜๋ฅผ ์ค„์ž…๋‹ˆ๋‹ค. ์ด ๊ณผ์ •์„ ์—ฌ๋Ÿฌ ๋ฒˆ ๋ฐ˜๋ณตํ•จ์œผ๋กœ์จ ๋ชจ๋ธ์€ ์ ์ฐจ ์ •ํ™•๋„๋ฅผ ํ–ฅ์ƒ์‹œํ‚ต๋‹ˆ๋‹ค. ์‹œ๊ฐ„์ด ์ง€๋‚จ์— ๋”ฐ๋ผ ๋ชจ์–‘, ์ƒ‰์ƒ, ์งˆ๊ฐ๊ณผ ๊ฐ™์€ ๋ณต์žกํ•œ ํŒจํ„ด์„ ์ธ์‹ํ•˜๋Š” ๋ฐฉ๋ฒ•์„ ํ•™์Šตํ•ฉ๋‹ˆ๋‹ค.

์—ญ์ „ํŒŒ๋ž€ ๋ฌด์—‡์ธ๊ฐ€์š”?

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

์ด์ œ ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•  ๋•Œ ๋’ค์—์„œ ์–ด๋–ค ์ผ์ด ์ผ์–ด๋‚˜๋Š”์ง€ ์•Œ์•˜์œผ๋‹ˆ ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•  ๋•Œ ๊ณ ๋ คํ•ด์•ผ ํ•  ์‚ฌํ•ญ์„ ์‚ดํŽด๋ณด๊ฒ ์Šต๋‹ˆ๋‹ค.

๋Œ€๊ทœ๋ชจ ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ๋Œ€ํ•œ ๊ต์œก

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

๋ฐฐ์น˜ ํฌ๊ธฐ ๋ฐ GPU ํ™œ์šฉ๋„

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

YOLOv8 ์™€ ๊ด€๋ จํ•˜์—ฌ ๋‹ค์Œ๊ณผ ๊ฐ™์ด ์„ค์ •ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. batch_size ๋งค๊ฐœ๋ณ€์ˆ˜์˜ ๊ต์œก ๊ตฌ์„ฑ ๋ฅผ GPU ์šฉ๋Ÿ‰์— ๋งž๊ฒŒ ์„ค์ •ํ•˜์„ธ์š”. ๋˜ํ•œ batch=-1 ๋ฅผ ๊ต์œก ์Šคํฌ๋ฆฝํŠธ์— ์ถ”๊ฐ€ํ•˜๋ฉด ๋””๋ฐ”์ด์Šค์˜ ๊ธฐ๋Šฅ์— ๋”ฐ๋ผ ํšจ์œจ์ ์œผ๋กœ ์ฒ˜๋ฆฌํ•  ์ˆ˜ ์žˆ๋Š” ๋ฐฐ์น˜ ํฌ๊ธฐ๊ฐ€ ์ž๋™์œผ๋กœ ๊ฒฐ์ •๋ฉ๋‹ˆ๋‹ค. ๋ฐฐ์น˜ ํฌ๊ธฐ๋ฅผ ๋ฏธ์„ธ ์กฐ์ •ํ•˜๋ฉด GPU ๋ฆฌ์†Œ์Šค๋ฅผ ์ตœ๋Œ€ํ•œ ํ™œ์šฉํ•˜๊ณ  ์ „๋ฐ˜์ ์ธ ๊ต์œก ํ”„๋กœ์„ธ์Šค๋ฅผ ๊ฐœ์„ ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

ํ•˜์œ„ ์ง‘ํ•ฉ ๊ต์œก

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

YOLOv8 ์˜ ๊ฒฝ์šฐ ๋‹ค์Œ์„ ์‚ฌ์šฉํ•˜์—ฌ ํ•˜์œ„ ์ง‘ํ•ฉ ๊ต์œก์„ ์‰ฝ๊ฒŒ ๊ตฌํ˜„ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. fraction ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์„ค์ •ํ•ฉ๋‹ˆ๋‹ค. ์ด ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ์–ด๋Š ๋ถ€๋ถ„์„ ํ•™์Šต์— ์‚ฌ์šฉํ• ์ง€ ์ง€์ •ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด, ์„ค์ • fraction=0.1 ๋Š” ๋ฐ์ดํ„ฐ์˜ 10%์— ๋Œ€ํ•ด ๋ชจ๋ธ์„ ํ•™์Šต์‹œํ‚ต๋‹ˆ๋‹ค. ์ „์ฒด ๋ฐ์ดํ„ฐ ์ง‘ํ•ฉ์„ ์‚ฌ์šฉํ•˜์—ฌ ๋ชจ๋ธ์„ ํ•™์Šต์‹œํ‚ค๊ธฐ ์ „์— ์ด ๊ธฐ์ˆ ์„ ์‚ฌ์šฉํ•˜์—ฌ ๋น ๋ฅด๊ฒŒ ๋ฐ˜๋ณตํ•˜๊ณ  ๋ชจ๋ธ์„ ํŠœ๋‹ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ํ•˜์œ„ ์ง‘ํ•ฉ ํ•™์Šต์€ ๋น ๋ฅธ ์ง„ํ–‰์„ ๋•๊ณ  ์ž ์žฌ์ ์ธ ๋ฌธ์ œ๋ฅผ ์กฐ๊ธฐ์— ์‹๋ณ„ํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋ฉ๋‹ˆ๋‹ค.

๋‹ค์ค‘ ๊ทœ๋ชจ ๊ต์œก

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

์˜ˆ๋ฅผ ๋“ค์–ด YOLOv8 ์„ ํŠธ๋ ˆ์ด๋‹ํ•  ๋•Œ ๋ฉ€ํ‹ฐ์Šค์ผ€์ผ ํŠธ๋ ˆ์ด๋‹์„ ํ™œ์„ฑํ™”ํ•˜๋ ค๋ฉด scale ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์„ค์ •ํ•ฉ๋‹ˆ๋‹ค. ์ด ๋งค๊ฐœ๋ณ€์ˆ˜๋Š” ํ›ˆ๋ จ ์ด๋ฏธ์ง€์˜ ํฌ๊ธฐ๋ฅผ ์ง€์ •๋œ ๊ณ„์ˆ˜๋งŒํผ ์กฐ์ •ํ•˜์—ฌ ๋‹ค์–‘ํ•œ ๊ฑฐ๋ฆฌ์— ์žˆ๋Š” ๋ฌผ์ฒด๋ฅผ ์‹œ๋ฎฌ๋ ˆ์ด์…˜ํ•ฉ๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด scale=0.5 ์€ ์ด๋ฏธ์ง€ ํฌ๊ธฐ๋ฅผ ์ ˆ๋ฐ˜์œผ๋กœ ์ค„์ด๊ณ  scale=2.0 ๋กœ ์„ค์ •ํ•˜๋ฉด ๋‘ ๋ฐฐ๊ฐ€ ๋ฉ๋‹ˆ๋‹ค. ์ด ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ๊ตฌ์„ฑํ•˜๋ฉด ๋ชจ๋ธ์ด ๋‹ค์–‘ํ•œ ์ด๋ฏธ์ง€ ๋ฐฐ์œจ์„ ๊ฒฝํ—˜ํ•˜๊ณ  ๋‹ค์–‘ํ•œ ๊ฐ์ฒด ํฌ๊ธฐ์™€ ์‹œ๋‚˜๋ฆฌ์˜ค์—์„œ ๊ฐ์ง€ ๊ธฐ๋Šฅ์„ ํ–ฅ์ƒ์‹œํ‚ฌ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์บ์‹ฑ

์บ์‹ฑ์€ ๋จธ์‹ ๋Ÿฌ๋‹ ๋ชจ๋ธ ํ•™์Šต์˜ ํšจ์œจ์„ฑ์„ ํ–ฅ์ƒ์‹œํ‚ค๋Š” ์ค‘์š”ํ•œ ๊ธฐ์ˆ ์ž…๋‹ˆ๋‹ค. ์บ์‹ฑ์€ ์‚ฌ์ „ ์ฒ˜๋ฆฌ๋œ ์ด๋ฏธ์ง€๋ฅผ ๋ฉ”๋ชจ๋ฆฌ์— ์ €์žฅํ•จ์œผ๋กœ์จ GPU ์—์„œ ๋””์Šคํฌ์—์„œ ๋ฐ์ดํ„ฐ๊ฐ€ ๋กœ๋“œ๋˜๊ธฐ๋ฅผ ๊ธฐ๋‹ค๋ฆฌ๋Š” ์‹œ๊ฐ„์„ ์ค„์—ฌ์ค๋‹ˆ๋‹ค. ๋ชจ๋ธ์€ ๋””์Šคํฌ I/O ์ž‘์—…์œผ๋กœ ์ธํ•œ ์ง€์—ฐ ์—†์ด ์ง€์†์ ์œผ๋กœ ๋ฐ์ดํ„ฐ๋ฅผ ์ˆ˜์‹ ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

๊ต์œกํ•  ๋•Œ YOLOv8 ์„ ์‚ฌ์šฉํ•˜์—ฌ ์บ์‹ฑ์„ ์ œ์–ดํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. cache ๋งค๊ฐœ๋ณ€์ˆ˜์ž…๋‹ˆ๋‹ค:

  • cache=True: ๋ฐ์ดํ„ฐ ์„ธํŠธ ์ด๋ฏธ์ง€๋ฅผ RAM์— ์ €์žฅํ•˜์—ฌ ๊ฐ€์žฅ ๋น ๋ฅธ ์•ก์„ธ์Šค ์†๋„๋ฅผ ์ œ๊ณตํ•˜์ง€๋งŒ ๋ฉ”๋ชจ๋ฆฌ ์‚ฌ์šฉ๋Ÿ‰์ด ์ฆ๊ฐ€ํ•ฉ๋‹ˆ๋‹ค.
  • cache='disk': ์ด๋ฏธ์ง€๋ฅผ ๋””์Šคํฌ์— ์ €์žฅํ•˜๋ฉฐ, RAM๋ณด๋‹ค๋Š” ๋Š๋ฆฌ์ง€๋งŒ ๋งค๋ฒˆ ์ƒˆ ๋ฐ์ดํ„ฐ๋ฅผ ๋กœ๋“œํ•˜๋Š” ๊ฒƒ๋ณด๋‹ค ๋น ๋ฆ…๋‹ˆ๋‹ค.
  • cache=False: ์บ์‹ฑ์„ ๋น„ํ™œ์„ฑํ™”ํ•˜์—ฌ ๊ฐ€์žฅ ๋Š๋ฆฐ ์˜ต์…˜์ธ ๋””์Šคํฌ I/O์—๋งŒ ์˜์กดํ•ฉ๋‹ˆ๋‹ค.

ํ˜ผํ•ฉ ์ •๋ฐ€ ๊ต์œก

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

ํ˜ผํ•ฉ ์ •๋ฐ€ ๊ต์œก ๊ฐœ์š”

ํ˜ผํ•ฉ ์ •๋ฐ€๋„ ํ›ˆ๋ จ์„ ๊ตฌํ˜„ํ•˜๋ ค๋ฉด ํ›ˆ๋ จ ์Šคํฌ๋ฆฝํŠธ๋ฅผ ์ˆ˜์ •ํ•˜๊ณ  ํ•˜๋“œ์›จ์–ด(์˜ˆ: GPU)๊ฐ€ ์ด๋ฅผ ์ง€์›ํ•˜๋Š”์ง€ ํ™•์ธํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค. Tensorflow ๊ณผ ๊ฐ™์€ ๋งŽ์€ ์ตœ์‹  ๋”ฅ๋Ÿฌ๋‹ ํ”„๋ ˆ์ž„์›Œํฌ๋Š” ํ˜ผํ•ฉ ์ •๋ฐ€๋„๋ฅผ ๊ธฐ๋ณธ์ ์œผ๋กœ ์ง€์›ํ•ฉ๋‹ˆ๋‹ค.

ํ˜ผํ•ฉ ์ •๋ฐ€๋„ ๊ต์œก์€ YOLOv8 ์œผ๋กœ ์ž‘์—…ํ•  ๋•Œ ๊ฐ„๋‹จํ•ฉ๋‹ˆ๋‹ค. amp ํ”Œ๋ž˜๊ทธ๋ฅผ ์„ค์ •ํ•ฉ๋‹ˆ๋‹ค. ์„ค์ • amp=True ๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ์ž๋™ ํ˜ผํ•ฉ ์ •๋ฐ€๋„(AMP) ํ›ˆ๋ จ์ด ๊ฐ€๋Šฅํ•ฉ๋‹ˆ๋‹ค. ํ˜ผํ•ฉ ์ •๋ฐ€๋„ ํ›ˆ๋ จ์€ ๋ชจ๋ธ ํ›ˆ๋ จ ํ”„๋กœ์„ธ์Šค๋ฅผ ์ตœ์ ํ™”ํ•˜๋Š” ๊ฐ„๋‹จํ•˜๋ฉด์„œ๋„ ํšจ๊ณผ์ ์ธ ๋ฐฉ๋ฒ•์ž…๋‹ˆ๋‹ค.

์‚ฌ์ „ ํ•™์Šต๋œ ๊ฐ€์ค‘์น˜

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

๊ทธ๋ฆฌ๊ณ  pretrained ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด YOLOv8 ์œผ๋กœ ์‰ฝ๊ฒŒ ์ „์ด ํ•™์Šต์„ ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์„ค์ • pretrained=True ๋Š” ๊ธฐ๋ณธ ์‚ฌ์ „ ํ•™์Šต๋œ ๊ฐ€์ค‘์น˜๋ฅผ ์‚ฌ์šฉํ•˜๊ฑฐ๋‚˜ ์‚ฌ์šฉ์ž ์ง€์ • ์‚ฌ์ „ ํ•™์Šต๋œ ๋ชจ๋ธ์˜ ๊ฒฝ๋กœ๋ฅผ ์ง€์ •ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์‚ฌ์ „ ํ•™์Šต๋œ ๊ฐ€์ค‘์น˜์™€ ์ „์ด ํ•™์Šต์„ ์‚ฌ์šฉํ•˜๋ฉด ๋ชจ๋ธ์˜ ๊ธฐ๋Šฅ์„ ํšจ๊ณผ์ ์œผ๋กœ ํ–ฅ์ƒ์‹œํ‚ค๊ณ  ํ•™์Šต ๋น„์šฉ์„ ์ ˆ๊ฐํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

๋Œ€๊ทœ๋ชจ ๋ฐ์ดํ„ฐ ์ง‘ํ•ฉ์„ ์ฒ˜๋ฆฌํ•  ๋•Œ ๊ณ ๋ คํ•ด์•ผ ํ•  ๊ธฐํƒ€ ๊ธฐ์ˆ 

๋Œ€๊ทœ๋ชจ ๋ฐ์ดํ„ฐ ์ง‘ํ•ฉ์„ ์ฒ˜๋ฆฌํ•  ๋•Œ ๊ณ ๋ คํ•ด์•ผ ํ•  ๋ช‡ ๊ฐ€์ง€ ๋‹ค๋ฅธ ๊ธฐ์ˆ ์ด ์žˆ์Šต๋‹ˆ๋‹ค:

  • ํ•™์Šต ์†๋„ ์Šค์ผ€์ค„๋Ÿฌ: ํ•™์Šต ์†๋„ ์Šค์ผ€์ค„๋Ÿฌ๋ฅผ ๊ตฌํ˜„ํ•˜๋ฉด ํ•™์Šต ์ค‘์— ํ•™์Šต ์†๋„๋ฅผ ๋™์ ์œผ๋กœ ์กฐ์ •ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ํ•™์Šต ์†๋„๋ฅผ ์ž˜ ์กฐ์ •ํ•˜๋ฉด ๋ชจ๋ธ์˜ ์ตœ์†Œ๊ฐ’ ์˜ค๋ฒ„์ŠˆํŒ…์„ ๋ฐฉ์ง€ํ•˜๊ณ  ์•ˆ์ •์„ฑ์„ ํ–ฅ์ƒ์‹œํ‚ฌ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. YOLOv8 ์„ ํ›ˆ๋ จํ•  ๋•Œ lrf ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ์ตœ์ข… ํ•™์Šต๋ฅ ์„ ์ดˆ๊ธฐ ํ•™์Šต๋ฅ ์˜ ์ผ๋ถ€๋ถ„์œผ๋กœ ์„ค์ •ํ•˜์—ฌ ํ•™์Šต๋ฅ  ์Šค์ผ€์ค„์„ ๊ด€๋ฆฌํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
  • ๋ถ„์‚ฐ ํ›ˆ๋ จ: ๋Œ€๊ทœ๋ชจ ๋ฐ์ดํ„ฐ ์„ธํŠธ๋ฅผ ์ฒ˜๋ฆฌํ•  ๋•Œ ๋ถ„์‚ฐ ํ›ˆ๋ จ์€ ํŒ๋„๋ฅผ ๋ฐ”๊ฟ€ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ํŠธ๋ ˆ์ด๋‹ ์›Œํฌ๋กœ๋“œ๋ฅผ ์—ฌ๋Ÿฌ GPU ๋˜๋Š” ๋จธ์‹ ์— ๋ถ„์‚ฐํ•˜์—ฌ ํŠธ๋ ˆ์ด๋‹ ์‹œ๊ฐ„์„ ๋‹จ์ถ•ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

ํ›ˆ๋ จํ•  ์—ํฌํฌ ์ˆ˜

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

ํ”ํžˆ ์ œ๊ธฐ๋˜๋Š” ์งˆ๋ฌธ์€ ๋ชจ๋ธ์„ ํ•™์Šต์‹œํ‚ฌ ์—ํฌํฌ ์ˆ˜๋ฅผ ๊ฒฐ์ •ํ•˜๋Š” ๋ฐฉ๋ฒ•์ž…๋‹ˆ๋‹ค. ์ข‹์€ ์‹œ์ž‘์ ์€ 300๊ฐœ ์—ํฌํฌ์ž…๋‹ˆ๋‹ค. ๋ชจ๋ธ์ด ์กฐ๊ธฐ์— ๊ณผ์ ํ•ฉํ•˜๋ฉด ์—ํฌํฌ ์ˆ˜๋ฅผ ์ค„์ผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. 300๊ฐœ ์—ํฌํฌ ์ดํ›„์—๋„ ๊ณผ์ ํ•ฉ์ด ๋ฐœ์ƒํ•˜์ง€ ์•Š์œผ๋ฉด 600๊ฐœ, 1200๊ฐœ ๋˜๋Š” ๊ทธ ์ด์ƒ์œผ๋กœ ํ›ˆ๋ จ์„ ํ™•์žฅํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

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

์กฐ๊ธฐ ์ค‘์ง€

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

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

์กฐ๊ธฐ ์ข…๋ฃŒ ๊ฐœ์š”

YOLOv8 ์˜ ๊ฒฝ์šฐ ํŠธ๋ ˆ์ด๋‹ ๊ตฌ์„ฑ์—์„œ ์ธ๋‚ด์‹ฌ ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์„ค์ •ํ•˜์—ฌ ์กฐ๊ธฐ ์ค‘์ง€๋ฅผ ํ™œ์„ฑํ™”ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด patience=5 ๋Š” 5ํšŒ ์—ฐ์†์œผ๋กœ ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ์ง€ํ‘œ๊ฐ€ ๊ฐœ์„ ๋˜์ง€ ์•Š์œผ๋ฉด ํŠธ๋ ˆ์ด๋‹์ด ์ค‘๋‹จ๋จ์„ ์˜๋ฏธํ•ฉ๋‹ˆ๋‹ค. ์ด ๋ฐฉ๋ฒ•์„ ์‚ฌ์šฉํ•˜๋ฉด ํŠธ๋ ˆ์ด๋‹ ํ”„๋กœ์„ธ์Šค๊ฐ€ ํšจ์œจ์ ์œผ๋กœ ์œ ์ง€๋˜๊ณ  ๊ณผ๋„ํ•œ ๊ณ„์‚ฐ ์—†์ด ์ตœ์ ์˜ ์„ฑ๋Šฅ์„ ๋‹ฌ์„ฑํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

ํด๋ผ์šฐ๋“œ ๊ต์œก๊ณผ ๋กœ์ปฌ ๊ต์œก ์ค‘ ์„ ํƒ

๋ชจ๋ธ ํŠธ๋ ˆ์ด๋‹์—๋Š” ํด๋ผ์šฐ๋“œ ํŠธ๋ ˆ์ด๋‹๊ณผ ๋กœ์ปฌ ํŠธ๋ ˆ์ด๋‹์˜ ๋‘ ๊ฐ€์ง€ ์˜ต์…˜์ด ์žˆ์Šต๋‹ˆ๋‹ค.

ํด๋ผ์šฐ๋“œ ํŠธ๋ ˆ์ด๋‹์€ ํ™•์žฅ์„ฑ๊ณผ ๊ฐ•๋ ฅํ•œ ํ•˜๋“œ์›จ์–ด๋ฅผ ์ œ๊ณตํ•˜๋ฉฐ ๋Œ€๊ทœ๋ชจ ๋ฐ์ดํ„ฐ ์„ธํŠธ์™€ ๋ณต์žกํ•œ ๋ชจ๋ธ์„ ์ฒ˜๋ฆฌํ•˜๋Š” ๋ฐ ์ด์ƒ์ ์ž…๋‹ˆ๋‹ค. Google Cloud, AWS, Azure์™€ ๊ฐ™์€ ํ”Œ๋žซํผ์€ ๊ณ ์„ฑ๋Šฅ GPU์™€ TPU์— ๋Œ€ํ•œ ์˜จ๋””๋งจ๋“œ ์•ก์„ธ์Šค๋ฅผ ์ œ๊ณตํ•˜์—ฌ ํŠธ๋ ˆ์ด๋‹ ์‹œ๊ฐ„์„ ๋‹จ์ถ•ํ•˜๊ณ  ๋Œ€๊ทœ๋ชจ ๋ชจ๋ธ์„ ์‹คํ—˜ํ•  ์ˆ˜ ์žˆ๋„๋ก ์ง€์›ํ•ฉ๋‹ˆ๋‹ค. ํ•˜์ง€๋งŒ ํด๋ผ์šฐ๋“œ ํŠธ๋ ˆ์ด๋‹์€ ํŠนํžˆ ์žฅ๊ธฐ๊ฐ„์— ๊ฑธ์ณ ๋น„์šฉ์ด ๋งŽ์ด ๋“ค ์ˆ˜ ์žˆ์œผ๋ฉฐ ๋ฐ์ดํ„ฐ ์ „์†ก์œผ๋กœ ์ธํ•ด ๋น„์šฉ๊ณผ ์ง€์—ฐ ์‹œ๊ฐ„์ด ๋Š˜์–ด๋‚  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

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

์ตœ์ ํ™” ๋„๊ตฌ ์„ ํƒ

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

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

์ผ๋ฐ˜์ ์ธ ์ตœ์ ํ™” ๋„๊ตฌ

์ตœ์ ํ™” ๋„๊ตฌ๋งˆ๋‹ค ๋‹ค์–‘ํ•œ ์žฅ๋‹จ์ ์ด ์žˆ์Šต๋‹ˆ๋‹ค. ๋ช‡ ๊ฐ€์ง€ ์ผ๋ฐ˜์ ์ธ ์ตœ์ ํ™” ๋„๊ตฌ์— ๋Œ€ํ•ด ๊ฐ„๋žตํžˆ ์‚ดํŽด๋ณด๊ฒ ์Šต๋‹ˆ๋‹ค.

  • SGD(ํ™•๋ฅ ์  ๊ทธ๋ผ๋ฐ์ด์…˜ ํ•˜๊ฐ•):

    • ๋งค๊ฐœ๋ณ€์ˆ˜์— ๋Œ€ํ•œ ์†์‹ค ํ•จ์ˆ˜์˜ ๊ธฐ์šธ๊ธฐ๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ๋ชจ๋ธ ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์—…๋ฐ์ดํŠธํ•ฉ๋‹ˆ๋‹ค.
    • ๊ฐ„๋‹จํ•˜๊ณ  ํšจ์œจ์ ์ด์ง€๋งŒ ์ˆ˜๋ ด ์†๋„๊ฐ€ ๋Š๋ฆฌ๊ณ  ๋กœ์ปฌ ์ตœ์†Œ๊ฐ’์— ๊ฐ‡ํž ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
  • ์•„๋‹ด(์ ์‘ํ˜• ์ˆœ๊ฐ„ ์ถ”์ •):

    • SGD์™€ ๋ชจ๋ฉ˜ํ…€ ๋ฐ RMSProp์˜ ์žฅ์ ์„ ๊ฒฐํ•ฉํ•ฉ๋‹ˆ๋‹ค.
    • ๊ธฐ์šธ๊ธฐ์˜ ์ฒซ ๋ฒˆ์งธ ๋ฐ ๋‘ ๋ฒˆ์งธ ๋ชจ๋ฉ˜ํŠธ์˜ ์ถ”์ •์น˜๋ฅผ ๊ธฐ๋ฐ˜์œผ๋กœ ๊ฐ ํŒŒ๋ผ๋ฏธํ„ฐ์˜ ํ•™์Šต ์†๋„๋ฅผ ์กฐ์ •ํ•ฉ๋‹ˆ๋‹ค.
    • ๋…ธ์ด์ฆˆ๊ฐ€ ๋งŽ์€ ๋ฐ์ดํ„ฐ์™€ ํฌ๋ฐ•ํ•œ ๊ทธ๋ผ๋ฐ์ด์…˜์— ์ ํ•ฉํ•ฉ๋‹ˆ๋‹ค.
    • ํšจ์œจ์ ์ด๊ณ  ์ผ๋ฐ˜์ ์œผ๋กœ ํŠœ๋‹์ด ๋œ ํ•„์š”ํ•˜๋ฏ€๋กœ YOLOv8 ์— ๊ถŒ์žฅ๋˜๋Š” ์ตœ์ ํ™” ํ”„๋กœ๊ทธ๋žจ์ž…๋‹ˆ๋‹ค.
  • RMSProp(๋ฃจํŠธ ํ‰๊ท  ์ œ๊ณฑ ์ „ํŒŒ):

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

YOLOv8 ์˜ ๊ฒฝ์šฐ optimizer ํŒŒ๋ผ๋ฏธํ„ฐ๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด SGD, Adam, AdamW, NAdam, RAdam, RMSProp ๋“ฑ ๋‹ค์–‘ํ•œ ์ตœ์ ํ™” ๋„๊ตฌ ์ค‘์—์„œ ์„ ํƒํ•˜๊ฑฐ๋‚˜ ๋‹ค์Œ๊ณผ ๊ฐ™์ด ์„ค์ •ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. auto ๋ฅผ ํด๋ฆญํ•˜์—ฌ ๋ชจ๋ธ ๊ตฌ์„ฑ์— ๋”ฐ๋ผ ์ž๋™์œผ๋กœ ์„ ํƒํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์ปค๋ฎค๋‹ˆํ‹ฐ์™€ ์—ฐ๊ฒฐํ•˜๊ธฐ

์ปดํ“จํ„ฐ ๋น„์ „ ์• ํ˜ธ๊ฐ€ ์ปค๋ฎค๋‹ˆํ‹ฐ์˜ ์ผ์›์ด ๋˜๋ฉด ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๊ณ  ๋” ๋นจ๋ฆฌ ๋ฐฐ์šฐ๋Š” ๋ฐ ๋„์›€์ด ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋‹ค์Œ์€ ์—ฐ๊ฒฐํ•˜๊ณ , ๋„์›€์„ ๋ฐ›๊ณ , ์•„์ด๋””์–ด๋ฅผ ๊ณต์œ ํ•˜๋Š” ๋ช‡ ๊ฐ€์ง€ ๋ฐฉ๋ฒ•์ž…๋‹ˆ๋‹ค.

์ปค๋ฎค๋‹ˆํ‹ฐ ๋ฆฌ์†Œ์Šค

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

๊ณต์‹ ๋ฌธ์„œ

  • Ultralytics YOLOv8 ๋ฌธ์„œ: ๊ณต์‹ ๋ฌธ์„œ( YOLOv8 )์—์„œ ๋‹ค์–‘ํ•œ ์ปดํ“จํ„ฐ ๋น„์ „ ํ”„๋กœ์ ํŠธ์— ๋Œ€ํ•œ ์ž์„ธํ•œ ๊ฐ€์ด๋“œ์™€ ์œ ์šฉํ•œ ํŒ์„ ํ™•์ธํ•˜์„ธ์š”.

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

์ฃผ์š” ๋‚ด์šฉ

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

์ž์ฃผ ๋ฌป๋Š” ์งˆ๋ฌธ

Ultralytics YOLO ๋กœ ๋Œ€๊ทœ๋ชจ ๋ฐ์ดํ„ฐ ์ง‘ํ•ฉ์„ ํ•™์Šตํ•  ๋•Œ GPU ํ™œ์šฉ๋„๋ฅผ ๋†’์ด๋ ค๋ฉด ์–ด๋–ป๊ฒŒ ํ•ด์•ผ ํ•˜๋‚˜์š”?

GPU ํ™œ์šฉ๋„๋ฅผ ๋†’์ด๋ ค๋ฉด batch_size ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ GPU ์—์„œ ์ง€์›ํ•˜๋Š” ์ตœ๋Œ€ ํฌ๊ธฐ๋กœ ์„ค์ •ํ•˜์„ธ์š”. ์ด๋ ‡๊ฒŒ ํ•˜๋ฉด GPU ์˜ ๊ธฐ๋Šฅ์„ ์ตœ๋Œ€ํ•œ ํ™œ์šฉํ•˜์—ฌ ํŠธ๋ ˆ์ด๋‹ ์‹œ๊ฐ„์„ ๋‹จ์ถ•ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋ฉ”๋ชจ๋ฆฌ ์˜ค๋ฅ˜๊ฐ€ ๋ฐœ์ƒํ•˜๋ฉด ํŠธ๋ ˆ์ด๋‹์ด ์›ํ™œํ•˜๊ฒŒ ์‹คํ–‰๋  ๋•Œ๊นŒ์ง€ ๋ฐฐ์น˜ ํฌ๊ธฐ๋ฅผ ์ ์ง„์ ์œผ๋กœ ์ค„์ด์„ธ์š”. YOLOv8 ์˜ ๊ฒฝ์šฐ ์„ค์ • batch=-1 ๋ฅผ ๊ต์œก ์Šคํฌ๋ฆฝํŠธ์— ์ถ”๊ฐ€ํ•˜๋ฉด ํšจ์œจ์ ์ธ ์ฒ˜๋ฆฌ๋ฅผ ์œ„ํ•œ ์ตœ์ ์˜ ๋ฐฐ์น˜ ํฌ๊ธฐ๊ฐ€ ์ž๋™์œผ๋กœ ๊ฒฐ์ •๋ฉ๋‹ˆ๋‹ค. ์ž์„ธํ•œ ๋‚ด์šฉ์€ ๊ต์œก ๊ตฌ์„ฑ.

ํ˜ผํ•ฉ ์ •๋ฐ€ ํ›ˆ๋ จ์ด๋ž€ ๋ฌด์—‡์ด๋ฉฐ YOLOv8 ์—์„œ ์–ด๋–ป๊ฒŒ ํ™œ์„ฑํ™”ํ•˜๋‚˜์š”?

ํ˜ผํ•ฉ ์ •๋ฐ€๋„ ํ›ˆ๋ จ์€ 16๋น„ํŠธ(FP16) ๋ฐ 32๋น„ํŠธ(FP32) ๋ถ€๋™ ์†Œ์ˆ˜์  ์œ ํ˜•์„ ๋ชจ๋‘ ํ™œ์šฉํ•˜์—ฌ ๊ณ„์‚ฐ ์†๋„์™€ ์ •๋ฐ€๋„์˜ ๊ท ํ˜•์„ ๋งž์ถฅ๋‹ˆ๋‹ค. ์ด ์ ‘๊ทผ ๋ฐฉ์‹์€ ๋ชจ๋ธ ์ •ํ™•๋„์˜ ์ €ํ•˜ ์—†์ด ํ•™์Šต ์†๋„๋ฅผ ๋†’์ด๊ณ  ๋ฉ”๋ชจ๋ฆฌ ์‚ฌ์šฉ๋Ÿ‰์„ ์ค„์ž…๋‹ˆ๋‹ค. YOLOv8 ์—์„œ ํ˜ผํ•ฉ ์ •๋ฐ€๋„ ํ›ˆ๋ จ์„ ์‚ฌ์šฉํ•˜๋„๋ก ์„ค์ •ํ•˜๋ ค๋ฉด amp ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ True ์„ ํŠธ๋ ˆ์ด๋‹ ๊ตฌ์„ฑ์— ์ถ”๊ฐ€ํ•ฉ๋‹ˆ๋‹ค. ์ด๋ ‡๊ฒŒ ํ•˜๋ฉด ์ž๋™ ํ˜ผํ•ฉ ์ •๋ฐ€๋„(AMP) ํ›ˆ๋ จ์ด ํ™œ์„ฑํ™”๋ฉ๋‹ˆ๋‹ค. ์ด ์ตœ์ ํ™” ๊ธฐ๋ฒ•์— ๋Œ€ํ•œ ์ž์„ธํ•œ ๋‚ด์šฉ์€ ๊ต์œก ๊ตฌ์„ฑ.

๋ฉ€ํ‹ฐ์Šค์ผ€์ผ ํŠธ๋ ˆ์ด๋‹์€ YOLOv8 ๋ชจ๋ธ ์„ฑ๋Šฅ์„ ์–ด๋–ป๊ฒŒ ํ–ฅ์ƒํ•˜๋‚˜์š”?

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

YOLOv8 ์—์„œ ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ์›จ์ดํŠธ๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ํ›ˆ๋ จ ์†๋„๋ฅผ ๋†’์ด๋ ค๋ฉด ์–ด๋–ป๊ฒŒ ํ•ด์•ผ ํ•˜๋‚˜์š”?

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

์—ํฌํฌ ์ˆ˜๋Š” ๋ชจ๋ธ ํ•™์Šต ์ค‘ ํ•™์Šต ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ์ „์ฒด ํ†ต๊ณผ ํšŸ์ˆ˜๋ฅผ ๋‚˜ํƒ€๋ƒ…๋‹ˆ๋‹ค. ์ผ๋ฐ˜์ ์ธ ์‹œ์ž‘์ ์€ 300๊ฐœ ์—ํฌํฌ์ž…๋‹ˆ๋‹ค. ๋ชจ๋ธ์ด ์กฐ๊ธฐ์— ๊ณผ์ ํ•ฉํ•˜๋ฉด ๊ทธ ์ˆ˜๋ฅผ ์ค„์ผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋˜๋Š” ๊ณผ์ ํ•ฉ์ด ๊ด€์ฐฐ๋˜์ง€ ์•Š๋Š” ๊ฒฝ์šฐ 600๊ฐœ, 1200๊ฐœ ๋˜๋Š” ๊ทธ ์ด์ƒ์˜ ์—ํฌํฌ๋กœ ํ›ˆ๋ จ์„ ํ™•์žฅํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์ด๋ฅผ ์„ค์ •ํ•˜๋ ค๋ฉด YOLOv8 ์—์„œ epochs ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์„ค์ •ํ•˜์„ธ์š”. ์ด์ƒ์ ์ธ ์—ํฌํฌ ์ˆ˜๋ฅผ ๊ฒฐ์ •ํ•˜๋Š” ๋ฐฉ๋ฒ•์— ๋Œ€ํ•œ ์ถ”๊ฐ€ ๋„์›€๋ง์€ ๋‹ค์Œ ์„น์…˜์„ ์ฐธ์กฐํ•˜์„ธ์š”. ์—ํฌํฌ ์ˆ˜.



์ƒ์„ฑ 2024-06-26, ์—…๋ฐ์ดํŠธ 2024-07-05
์ž‘์„ฑ์ž: glenn-jocher (3), Laughing-q (1), abirami-vina (1)

๋Œ“๊ธ€