boostcampaitech4lv23cv3 / final-project-level3-cv-17

final-project-level2-cv-17 created by GitHub Classroom
2 stars 0 forks source link
boostcamp-ai-tech computer-vision object-detection

๐ŸŒ  Sixth Sense ํŒ€์˜ Final ํ”„๋กœ์ ํŠธ

๐Ÿ˜Ž Members

์„œ์žฅ์› ๋ฐ•์„ ๊ทœ ๋ฐ•์„ธ์ค€ ์žฅ๊ตญ๋นˆ
@nanpuhaha @Sungyu-Park @sjleo1 @JKbin

๐Ÿง‘โ€๐Ÿ’ป Contributions

:car: Project Overview

์—ฌ๋Ÿฌ๋ถ„์€ ์ตœ๊ทผ ์šฐ๋ฆฌ๋‚˜๋ผ์—์„œ ๋ถˆ๊ฑฐ์ง„ ๋ช‡๋ช‡ ๊ตฌ๊ธ‰์ฐจ ๊ธธ๋ง‰ ์‚ฌ๊ฑด์„ ๊ธฐ์–ตํ•˜๊ณ  ๊ณ„์‹ ๊ฐ€์š”?

์ด๋Ÿฐ ์‚ฌ๊ฑด๋“ค์€ ์‚ฌ๋žŒ์˜ ์˜๋„์ ์ธ ํ–‰๋™์—์„œ ๋น„๋กฏ๋˜์—ˆ์Šต๋‹ˆ๋‹ค. ํ•˜์ง€๋งŒ ์ž์œจ์ฃผํ–‰ ์†Œํ”„ํŠธ์›จ์–ด๋Š” ๊ตฌ๊ธ‰์ฐจ์™€ ์†Œ๋ฐฉ์ฐจ, ๊ฒฝ์ฐฐ์ฐจ์™€ ๊ฐ™์€ ๊ธด๊ธ‰์ฐจ๋Ÿ‰์„ ๊ตฌ๋ถ„ํ•˜์ง€ ๋ชปํ•œ๋‹ค๋ฉด ์„ ํƒ์˜ ์—ฌ์ง€ ์—†์ด ๊ธธ์„ ๋น„์ผœ ์ฃผ์ง€ ๋ชปํ•  ๊ฒƒ์ž…๋‹ˆ๋‹ค.

์ž์œจ์ฃผํ–‰ ์ฐจ๋Ÿ‰์ด ๊ธด๊ธ‰์ฐจ๋Ÿ‰์„ ์œ„ํ•ด ํšŒํ”ผ ๊ฒฝ๋กœ๋ฅผ ํƒ์ƒ‰ํ•˜๊ณ  ์ฐจ๋Ÿ‰์„ ์ œ์–ดํ•˜๊ธฐ ์œ„ํ•ด์„œ๋Š” ๊ธด๊ธ‰์ฐจ๋Ÿ‰์ธ์ง€๋ฅผ ํŒ๋‹จํ•˜๋Š” ๊ณผ์ •์ด ๋จผ์ € ์ด๋ฃจ์–ด์ ธ์•ผ ํ•˜๊ธฐ ๋•Œ๋ฌธ์ž…๋‹ˆ๋‹ค.

๋”ฐ๋ผ์„œ ์ €ํฌ๋Š” ์ž์œจ์ฃผํ–‰ ์†Œํ”„ํŠธ์›จ์–ด์˜ ํ•ต์‹ฌ์ ์ธ ํ•œ ๋ถ€๋ถ„์œผ๋กœ์„œ ํ•ด๋‹น ์ฃผ์ œ๋ฅผ ์„ ์ •ํ•˜์˜€์Šต๋‹ˆ๋‹ค.

์ด๋ฟ๋งŒ ์•„๋‹ˆ๋ผ ์ €ํฌ๊ฐ€ ์ œ์ž‘ํ•œ ๋ชจ๋ธ์„ ๋‹ค๋ฅธ ๋ถ„์•ผ์—๋„ ์‘์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

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

์ €ํฌ๊ฐ€ ์ œ์ž‘ํ•œ ๋ชจ๋ธ๋กœ ์ด๋Ÿฌํ•œ ์‹œ์Šคํ…œ์„ ์ž๋™ํ™”ํ•˜์—ฌ ํ›จ์”ฌ ๋น ๋ฅด๊ณ  ์•ˆ์ „ํ•˜๋ฉฐ ๊ฒฝ์ œ์ ์œผ๋กœ ์šด์šฉํ•  ์ˆ˜ ์žˆ์„ ๊ฒƒ์ž…๋‹ˆ๋‹ค.

๐Ÿ’พ Dataset

  1. ์ฃผํ–‰ ์ฐจ๋Ÿ‰ ๊ด€์ ์˜ ํŠน์ˆ˜ ์ฐจ๋Ÿ‰ ํ˜•์ƒ ๋ฐ์ดํ„ฐ (AIHub)
  2. ์ž์œจ์ฃผํ–‰ ๋ฐ ADAS AI์ธ์ง€๋ชจ๋ธ ํ•™์Šต์šฉ ๋ฐ์ดํ„ฐ (AIHub)

๐Ÿงน Data Cleaning

ํ™•๋ณด ๋œ ์ด๋ฏธ์ง€ ๋ฐ์ดํ„ฐ์˜ ์ผ๊ด€์„ฑ๊ณผ ์ •ํ™•๋„๋ฅผ ์œ„ํ•ด Data Cleaning ์ž‘์—…์„ ์‹ค์‹œํ•˜์˜€์Šต๋‹ˆ๋‹ค. ๊ฒฐ๊ณผ ๋ผ๋ฒจ๋ง ์˜ค๋ฅ˜ 1,097๊ฑด / ์œก์•ˆ์œผ๋กœ ์‹๋ณ„์ด ๋ถˆ๊ฐ€๋Šฅ or ๊ฐ์ฒด๊ฐ€ 2/3 ์ด์ƒ ๊ฐ€๋ ค์ ธ ์žˆ๋Š” ๊ฒฝ์šฐ 1,324๊ฑด / ์–ด๋…ธํ…Œ์ด์…˜ ๋ถˆ๋Ÿ‰ 371๊ฑด์ด ๋ฐœ๊ฒฌ๋˜์—ˆ์Šต๋‹ˆ๋‹ค. ๋…ธ์ด์ฆˆ๊ฐ€ ์ด๋Š” ์ด๋ฏธ์ง€๊ฐ€ ์ „์ฒด ์ด๋ฏธ์ง€ ๋Œ€๋น„ ์•ฝ 16%๋ฅผ ์ฐจ์ง€ํ•˜์—ฌ ๋ผ๋ฒจ๋ง ์˜ค๋ฅ˜ ๋ฐ ์‹๋ณ„ ๋ถˆ๊ฐ€ ์ด๋ฏธ์ง€๋Š” ์‚ญ์ œ ํ•˜๊ณ  ์–ด๋…ธํ…Œ์ด์…˜ ๋ถˆ๋Ÿ‰ ์ด๋ฏธ์ง€๋Š” ์žฌ์–ด๋…ธํ…Œ์ด์…˜ ์ž‘์—…์„ ์ง„ํ–‰ํ•˜์˜€์Šต๋‹ˆ๋‹ค.

๐Ÿฆพ Model

์ž์œจ์ฃผํ–‰์— ์žˆ์–ด์„œ ์‹ค์‹œ๊ฐ„์œผ๋กœ ์ •ํ™•ํ•˜๊ฒŒ ๊ฐ์ฒด๋ฅผ ํƒ์ง€ํ•˜๋Š” ๊ฒƒ์€ ๋งค์šฐ ์ค‘์š”ํ•ฉ๋‹ˆ๋‹ค. ๋”ฐ๋ผ์„œ 2-Stage Detector๋ณด๋‹ค๋Š” 1-Stage Detector๋ฅผ ์‚ฌ์šฉํ•˜๋Š” ๊ฒƒ์ด ๋‚ซ๋‹ค๊ณ  ํŒ๋‹จํ–ˆ๊ณ , ์ด ์ค‘ ๋น ๋ฅด๊ฒŒ ๋ฐœ์ „ํ•˜๊ณ  ์žˆ๋Š” YOLO ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜๊ธฐ๋กœ ๊ฒฐ์ •ํ•˜์˜€์Šต๋‹ˆ๋‹ค.

๐Ÿ“ƒ Metric

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

โšก Service Develop

๐Ÿ“นDemo