작성
·
323
0
안녕하세요!
YOLO v1의 이해 - 01에서 바운딩 박스가 셀마다 2개식 생성이 되는데 이때 셀마다 갖는 바운딩 박스의 크기나 모양 등이 동일한 것인가요? 아니면 랜덤하게 생성이 되는것인가요?
감사합니다~
답변 1
1
안녕하십니까,
yolo v1 의 셀마다 예측하는 바운딩 박스는 anchor box와 개념이 좀 다릅니다. yolo v1의 바운딩 박스는 셀별로 학습이 되어서 계산이 되는 바운딩 박스입니다. 그래서 셀별로 바운딩 박스가 오브젝트를 예측하는 방식으로 모델이 학습이 되고, 새로운 데이터가 입력되었을 때 최적으로 예측된 바운딩 박스를 nms를 통해서 filtering하여 Object detection하는 방식입니다.
따라서 고정된 사이즈를 가지고 있지 않으며 모델을 학습 시켜서 만들어지게 됩니다.
감사합니다.
그러면 grid cell 마다 가지고 있는 bounding box가 IOU 값을 통해서 업데이트 된다는 말씀이신가요?