hye-log

[๋ถ€์ŠคํŠธ์บ ํ”„ AI Tech]WEEK 09_DAY 41 ๋ณธ๋ฌธ

Boostcourse/AI Tech 4๊ธฐ

[๋ถ€์ŠคํŠธ์บ ํ”„ AI Tech]WEEK 09_DAY 41

iihye_ 2022. 11. 16. 22:25

๐ŸŒฟ ๊ฐœ๋ณ„ํ•™์Šต


[3] Object Detection Library

1. Object Detection ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ

1) MMDetection

- ์ „์ฒด ํ”„๋ ˆ์ž„์›Œํฌ๋ฅผ ๋ชจ๋“ˆ ๋‹จ์œ„๋กœ ๋ถ„๋ฆฌํ•ด ๊ด€๋ฆฌํ•  ์ˆ˜ ์žˆ์Œ

- ๋งŽ์€ ํ”„๋ ˆ์ž„์›Œํฌ๋ฅผ ์ง€์›ํ•จ

- ๋‹ค๋ฅธ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ์— ๋น„ํ•ด ๋น ๋ฆ„

- Fast R-CNN, SSD, YOLO v3, DETR ๋“ฑ

2) Detectron2

- ์ „์ฒด ํ”„๋ ˆ์ž„์›Œํฌ๋ฅผ ๋ชจ๋“ˆ ๋‹จ์œ„๋กœ ๋ถ„๋ฆฌํ•ด ๊ด€๋ฆฌํ•  ์ˆ˜ ์žˆ์Œ

- OD ์™ธ์—๋„ Segmentation, Pose estimation ๋“ฑ์˜ ์•Œ๊ณ ๋ฆฌ์ฆ˜์„ ์ง€์›ํ•จ

- Faster R-CNN, RetinaNet, Mask R-CNN, DETR ๋“ฑ

 

2. MMDetection

1) Pytorch ๊ธฐ๋ฐ˜์˜ Object Detection ์˜คํ”ˆ์†Œ์Šค ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ

2) Pipeline

- Input -> Backbone -> Neck -> Dense Prediction -> Prediction

3) 2 Stage ๋ชจ๋ธ์€ Backbone / Neck / DenseHead / RoIHead ๋ชจ๋“ˆ๋กœ ๋‚˜๋ˆŒ ์ˆ˜ ์žˆ์Œ

4) Config File

- ๋ฐ์ดํ„ฐ์…‹, ๋ชจ๋ธ, schedular, optimizer ์ •์˜ ๊ฐ€๋Šฅ

 

3. Detectron2

1) Pytorch ๊ธฐ๋ฐ˜ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ

2) Object Detection ์™ธ์—๋„ Segmentation, Pose prediction ๋“ฑ ์•Œ๊ณ ๋ฆฌ์ฆ˜ ์ œ๊ณต

3) Pipeline

- Setup Config / Setup Trainer / Start Training



๐ŸŒฟ ์˜ค๋Š˜์˜ ํšŒ๊ณ 

์˜ค์ „์—๋Š” Object Detection ๊ฐ•์˜๋ฅผ ๋“ค์—ˆ๋‹ค. ์•„๋ฌด๋ž˜๋„ ๋Œ€ํšŒ๋ฅผ ์‹œ์ž‘ํ•˜๋ ค๋ฉด ๊ธฐ๋ณธ์ ์ธ ์ง€์‹์„ ์Šต๋“ํ•˜๋Š” ๊ฒŒ ์ค‘์š”ํ• ๊ฑฐ ๊ฐ™์•„์„œ ๊ฐ•์˜ ๋จผ์ € ๋น ๋ฅด๊ฒŒ ๋“ค์–ด๋ณด๋ ค๊ณ  ๋…ธ๋ ฅ์ค‘... ์˜คํ›„์—๋Š” ์šฐํŒ€์†Œ๊ฐ€ ์žˆ์—ˆ๋Š”๋ฐ ์ƒˆ๋กœ์šด ํŒ€๊ณผ ์–ด๋–ค ์ฃผ์ œ๋“ค๋กœ ์ตœ์ข… ํ”„๋กœ์ ํŠธ๋ฅผ ์ค€๋น„ํ•˜๊ณ  ์žˆ๋Š”์ง€ ๋‹ค๋ฅธ ํŒ€๋“ค์˜ ์ด์•ผ๊ธฐ๋ฅผ ๋“ค์–ด๋ณผ ์ˆ˜ ์žˆ์—ˆ๋‹ค. ์˜ค๋Š˜ ์ปดํผ๋‹ˆ ๋ฐ์ด ์žˆ๋Š” ์ค„ ์•Œ์•˜๋Š”๋ฐ ๋‹คํ–‰ํžˆ๋„ ๋‹ค์Œ ์ฃผ์˜€์Œ..!! ๊ฐ•์˜์™€ ์—ฐ๊ด€๋œ ์‹ค์Šตํ•˜๋‹ค๊ฐ€ training dataset ๊ฐ€์ง€๊ณ  ์‹œ๊ฐํ™” ํ•ด๋ณด๋ฉด ์ข‹์„๊ฑฐ ๊ฐ™์•„์„œ, wandb๋กœ ์‹œ๊ฐํ™”ํ•˜๋Š” ์ฝ”๋“œ๋ฅผ ์ž‘์„ฑํ•ด๋ณด์•˜๋‹ค. ์‚ฌ์‹ค ์•Œ๊ณ  ๋ณด๋‹ˆ train.json์ด ์•„๋‹ˆ๋ผ submission ํŒŒ์ผ๋กœ ์‹œ๊ฐํ™”ํ•ด์„œ ๊ฒฐ๊ณผ๊ฐ€ ์—‰๋šฑํ•˜๊ฒŒ ๋‚˜์™”๋‹ค..ใ…‹-ใ…‹ ํ”ผ์–ด์„ธ์…˜ ๋•Œ ์–ด๋–ค๊ฑฐ ํ•ด๋ณผ์ง€ ์ด์•ผ๊ธฐํ•˜๊ณ , ๋‹ค์‹œ ์ฝ”๋“œ๋ฅผ ์ฒ˜์Œ๋ถ€ํ„ฐ ์ž‘์„ฑํ•ด๋ณด๋‹ˆ json ํŒŒ์ด๋ ์„œ box ์ขŒํ‘œ ๊ฐ€์ ธ์™€์„œ wandb์— log๋กœ ๋ณด๋‚ด๊ณ  ๊ทธ๋ ค๋ณด์•˜๋‹ค..! ์‚ฌ์‹ค ์ž๊พธ ์•ˆ ๋˜์„œ ํฌ๊ธฐํ•˜๊ณ  ๋‹ค๋ฅธ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ์จ๋ณด๋ ค๊ณ  ํ–ˆ๋Š”๋ฐ log๋กœ ๋„˜๊ธธ ๋•Œ ์˜คํƒ€๊ฐ€ ์žˆ์—ˆ์–ด์„œ ์•ˆ ๋„˜์–ด๊ฐ€๋Š” ๊ฑฐ์˜€๋‹ค........(ใ… ) ์•„๋ฌดํŠผ ์„ฑ๊ณตํ–ˆ์œผ๋‹ˆ ์˜์˜๋ฅผ ๋‘๊ณ ... ์ฝ”์–ด ํƒ€์ž„ ๋๋‚˜๊ณ ๋Š” ๋ฉ˜ํ† ๋ง ์‹œ๊ฐ„์ด ์žˆ์—ˆ๋Š”๋ฐ ์ด๋ฒˆ ์ฃผ ํ•™์Šตํ•œ ๋‚ด์šฉ์— ๋Œ€ํ•ด์„œ ๊ฐ„๋‹จํ•œ overview์™€ object detection ๋Œ€ํšŒ ๊ด€๋ จ ํŒ์„ ์ฃผ์…จ๋‹ค..! object detection์€ ์ฒ˜์Œ ํ•ด๋ณด๋Š” ๋Œ€ํšŒ์ธ์ง€๋ผ ์‹œ์ž‘์„ ์–ด๋–ป๊ฒŒ ํ•ด์•ผํ• ์ง€ ๊ฐ๋„ ์ž˜ ์•ˆ ์˜ค๊ณ , ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ์˜ ์ฝ”๋“œ๋“ค๋„ ๋งŽ์•„์„œ (0-0) <-- ์ด ์ƒํƒœ์˜€๋Š”๋ฐ ์กฐ๊ธˆ์”ฉ ๊ธธ์ด ๋ณด์ด๋Š”๊ฑฐ ๊ฐ™๊ธฐ๋„ ํ•˜๋‹ค...

์ƒ๊ฐํ•ด๋ณด๋‹ˆ ๋ฐ์ดํ„ฐ์…‹์€ ๋น„๊ณต๊ฐœ๋ผ์„œ ๋Œ€์ถฉ bounding box ๊ทธ๋ฆฐ ๊ฑฐ์— ์˜์˜๋ฅผ ๋‘๊ธฐ....

728x90
Comments