본문 바로가기
딥러닝관련/Detection

Single shot detector에서 bounding box 예측(offset 및 SSD output based)

by 머리올리자 2022. 7. 4.

Prior box = (cx_p, cy_p, w_p, h_p)

SSD output(loc) : (Δcx, Δcy, Δw, Δh)

 

위 사전 정의한 prior box와 ssd output으로 bounding box 좌표를 예측할 수 있음.

 

cx = cx_p + 0.1Δcx * w_p

cy = cy_p + 0.1Δcy * h_p

w  = w_p + exp(0.2Δw)

h = h_p + exp(0.2Δh)

 

(위 0.1과 0.2는 variance 값들로 사전에 정의된다)


1) Positive box: Prior box 중 ground truth box와 IoU가 0.5 이상인 prior box (object 찾기)

2) Negative box: Prior box 중 ground truth box와 IoU가 0.5 미만인 prior box (background label 처리)

 

[Box prediction]

Prior box -> 정답 bounding box와의 IoU 측정 -> Prior box가 정답 box로 변형시키는 값들을 예측

(Δcx, Δcy, Δw, Δh)

 

[Classification]

confidence score를 통해 label 예측

'딥러닝관련 > Detection' 카테고리의 다른 글

Tutorial to Object Detection  (0) 2022.07.10
Jaccard 계수 (Jaccard coefficient)  (0) 2022.07.06
SSD - VGG backbone feature  (0) 2022.06.23
PASCAL VOC class 순서  (0) 2022.06.22
SSD(single shot detector) 코드 분석 1. (data)  (0) 2022.06.04