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

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

HAHA님의 프로필 이미지
HAHA

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

YOLO 실습부분 질문 있습니다.

작성

·

130

0

안녕하세요. 강사님

강사님 강의 YOLO - OpenCV DNN으로 YOLO Object Detection 실습 01 - 15:50 정도에 NMS 설명을 해주시는데 이해가 잘 안가는 부분이 있습니다.

NMS 할때 conf_threshold를 넣어주던데, 이 부분은 Detection 할때 confidence를 한번 필터링 하여 boxes와 confidences에 이미 0.5보다 큰 값들을 append해주어 사실 중복해서 하는거 아닌가요??

그러면 이 부분 코드는 opencv에서 다른 알고리즘에서도 사용하기위해서 인자를 받는것이고 YOLO를 사용할때는 2차 필터링을 위해 사용하는 정도로 생각하면 되는것인가요??

감사합니다!

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

네, 맞습니다. 중복으로 보셔도 무방합니다.

하지만 NMS전에 먼저 Filtering하는 코드를 쓰는게 좋습니다. 

NMS 구현 로직 별로 차이가 있을 수 있습니다만, 보통 Confidence score로 descending sort를 시킨 다음에 filtering out 을 시키는 경우가 있어서, 수행 시간을 조금이라도 단축하려면 먼저 filtering 하는게 좋습니다. 

감사합니다.

HAHA님의 프로필 이미지
HAHA

작성한 질문수

질문하기