Search

๐Ÿ‘ฉโ€๐Ÿ’ปTeam PlayCarnival

์ƒ์„ฑ์ผ
2023/08/25 10:10
ํƒœ๊ทธ
๊ธฐ๊ฐ„
2021.02 ~ 2021.10
๋‹ด๋‹น์—…๋ฌด
AI Engineer
์นดํ…Œ๊ณ ๋ฆฌ
์ฐฝ์—…๋™์•„๋ฆฌ

ํ’‹์‚ด ๊ฒฝ๊ธฐ ์ž๋™ ์ดฌ์˜ ๋ฐ ํŽธ์ง‘ ์‹œ์Šคํ…œ ๊ฐœ๋ฐœ

โ€ข
๊ด‘๊ฐ์นด๋ฉ”๋ผ 2๋Œ€ ๊ธฐ๊ณ„ ์–‘ ์˜†์„ ๋ฐ”๋ผ๋ณด๋„๋ก ์„ค์น˜
โ€ข
image stitching์„ ํ†ตํ•œ ๊ฒฝ๊ธฐ์žฅ ์ „์ฒด ์˜์ƒ ํš๋“
โ€ข
ํŽธ์ง‘ ์•Œ๊ณ ๋ฆฌ์ฆ˜ ์ ์šฉ์„ ํ†ตํ•ด ๋ฆฌํ”Œ๋ ˆ์ด ์˜์ƒ ์ƒ์„ฑ
โ€ข
์ดˆ๊ธฐ ๋ชจ๋ธ์€ ๋ผ์ฆˆ๋ฒ ๋ฆฌํŒŒ์ด๋ฅผ ๊ธฐ๋ฐ˜์œผ๋กœ ์„ค๊ณ„๋˜์—ˆ์œผ๋‚˜ ์ดํ›„ jetson nano ๋ณด๋“œ ๊ธฐ๋ฐ˜์œผ๋กœ ๋ณ€๊ฒฝ

1. Object Detection

YOLO-v5 ๋ชจ๋ธ ์ ์šฉ
backbone์œผ๋กœ CSP Darknet ์‚ฌ์šฉ
Darknet์— CSP๋ฅผ ์ ์šฉ
โ†’ CSP : feature fusion์ „๋žต์„ ํ†ตํ•ด transition layer๋ฅผ ์‚ฌ์šฉํ•จ์œผ๋กœ์จ ์—ฐ์‚ฐ๋Ÿ‰์„ ์ค„์ด๊ณ  bottleneck์„ ์‚ญ์ œ
object detection์˜ ์ „ํ†ต์ ์ธ ๋ฌธ์ œ์ธ ์ž‘์€ ๊ฐ์ฒด ์ธ์‹(๊ณต)์— ๋Œ€ํ•œ ์„ฑ๋Šฅ์ด ๋‚ฎ์€ ๋ฌธ์ œ
โ€ข
๋งŽ์€ ๋ฐ์ดํ„ฐ ํ™•๋ณด๊ฐ€ ์ตœ์šฐ์„ ์ด๋ผ๊ณ  ํŒ๋‹จ
1.
4000์žฅ์ •๋„์˜ ํฌ๋กค๋ง
2.
40000์žฅ ์ •๋„์˜ ์™ธ์ฃผ( ํ•œ์„ฑ๋Œ€ํ•™๊ต ํ’‹์‚ด์žฅ์—์„œ ์ดฌ์˜ํ•œ ์ƒ˜ํ”Œ์˜์ƒ)
โ†’ ๋น„์Šทํ•œ ๋ฐ์ดํ„ฐ๋“ค์ด ๋งŽ๊ณ , ๋‹ค์–‘ํ•œ ํ™˜๊ฒฝ์˜ ๋ฐ์ดํ„ฐ๊ฐ€ ํ•„์š”ํ•˜๋‹ค๊ณ  ํŒ๋‹จ(๋ฐค/๋‚ฎ, ์ž”๋””/ํ™ ๋“ฑ)
3.
ํฌ๋กค๋ง๊ณผ copy & paste augmentation

2. Tracking

Deep-SORT ์•Œ๊ณ ๋ฆฌ์ฆ˜ ํ™œ์šฉ
์ž‘์€ ๊ณต ๊ฐ์ฒด์— ๋Œ€ํ•œ ์„ฑ๋Šฅ ํ–ฅ์ƒ์€ ๊ณ„์†ํ•ด์„œ ๊ฐ€์ ธ๊ฐ€์•ผ ํ•˜๋Š” ๋ฌธ์ œ๋ผ๊ณ  ํŒ๋‹จ
์‚ฌ๋žŒ ๊ฐ์ฒด๋“ค์—๊ฒŒ Deep-SORT๋ฅผ ์ ์šฉํ•ด ์ด๋™๋Ÿ‰์„ ๊ณ„์‚ฐํ•ด ๋ฆฌํ”Œ๋ ˆ์ด ์˜์ƒ ์•Œ๊ณ ๋ฆฌ์ฆ˜์— ์ ์šฉํ•ด๋ณด๊ณ ์ž ํ•จ.
๊ฐ„๋‹จํ•˜๊ณ  ์„ฑ๋Šฅ์ด ์ข‹์€ ์•Œ๊ณ ๋ฆฌ์ฆ˜์ด์ง€๋งŒ, ID switching์ด ๋„ˆ๋ฌด ๋งŽ์ด ๋ฐœ์ƒํ–ˆ๋‹ค.
์šฐ๋ฆฌ์˜ ์˜์ƒ์€ ์‚ฌ๋žŒ์ด 13๋ช…์—์„œ ๋งŽ์•„๋ดค์ž 17~18๋ช…์˜ ์‚ฌ๋žŒ์ด ๋“ฑ์žฅํ•˜๋ฏ€๋กœ id switching์„ ๊ฐ€๋Šฅํ•œ ์ตœ์†Œํ™”
1.
track์˜ ์ˆ˜๋ช…์ธ max_age๋ฅผ ์˜์ƒ์ „์ฒด๋กœ ์„ค์ •ํ•ด ํ•œ๋ฒˆ ํ• ๋‹น๋œ track์€ ์ œ๊ฑฐ๋˜์ง€ ์•Š๋„๋ก ์„ค์ •
2.
track์„ ํ• ๋‹นํ•˜๋Š” ๊ธฐ์ค€์ด ๋˜๋Š” max_iou_distance๋ฅผ ๋‚ฎ์ถฐ๋ณด๋Š” ๋ฐฉ๋ฒ•( ํ• ๋‹น์ด ์กฐ๊ธˆ ๋” ๋„ˆ๊ทธ๋Ÿฝ๊ฒŒ ๋˜๋„๋ก)
์œ„์˜ ๋‘ ๋ฐฉ๋ฒ•์œผ๋กœ ์–ด๋Š์ •๋„ id switching์ด ๊ฐ์†Œ โ†’ idํ• ๋‹น์ด 990๋Œ€์—์„œ 200๋Œ€ ์ •๋„๋กœ ๊ฐ์†Œ
3.
id๋ฒˆํ˜ธ๋ฅผ 13์œผ๋กœ ์ œํ•œ : ์ƒ์„ฑ๋œ tracker์™€ ํ• ๋‹น ๊ธฐ์ค€์ด ๋งŒ์กฑ๋˜์ง€ ์•Š์•„๋„, ์ตœ๋Œ€ํ•œ state๊ฐ€ ๊ฐ€๊นŒ์šด ๊ฒƒ์œผ๋กœ detection๊ณผ tracker๋ฅผ ๋ฌด์กฐ๊ฑด ํ• ๋‹น
4.
๊ณต์˜ ์œ„์น˜ ๋ฐ ์›€์ง์ž„ ๋ฒกํ„ฐ์™€ ์‚ฌ๋žŒ๋“ค์˜ ์›€์ง์ž„ ๋ฒกํ„ฐ๋“ค์„ ๊ธฐ์ค€์œผ๋กœ ๊ฐ ํ”„๋ ˆ์ž„ crop ์˜์—ญ์„ ๊ฒฐ์ •