인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

김민준님의 프로필 이미지
김민준

작성한 질문수

YOLO 구현으로 배우는 딥러닝 논문 구현 with TensorFlow 2.0

Non-Maximum Suppression (NMS)

confidence가 가장 큰 값으로 IOU 비교 이후 과정에 대해

작성

·

293

0

안녕하세요. 좋은 강의 잘 보고 있습니다.

confidence 내림차순 정렬 이후 가장 큰 값으로 그보다 작은 값들의 IOU 계산으로 걸러낸 이후,

남은 값들 중 두번째로 큰 값으로 재차 IOU 계산을 하는 이유가 궁금합니다.

개인적인 이해로는 가장 큰 값을 구했다면 그것이 이미지내 object를 가장 잘 대표하는 box이니 이후 연산은 불필요하다고 생각되었습니다.

추가적으로 고민해봤는데, 아래 사진처럼

흰색 말 바로 오른쪽의 검은색 말을 그려주기 위해서 두번째로 큰 confidence로 nms과정을 거치는 것인가요??

답변 1

0

AISchool님의 프로필 이미지
AISchool
지식공유자

네. 안녕하세요~.

생각하신바가 맞습니다^^.

하나의 이미지내에 동일 class에 대한 여러개의 Object가 존재할수 있기 때문에 해당 과정을 적용해주는 것입니다.

감사합니다~.

김민준님의 프로필 이미지
김민준

작성한 질문수

질문하기