인프런 커뮤니티 질문&답변
작성한 질문수
detection 할 class 관해 질문이 있습니다.
작성
·
126
0
안녕하세요! 강사님의 강의를 잘 듣고 있습니다.
존윅 비디오에서 detection을 할때 tie 를 보면 0.9정도의 score를 통해 탐지가 되는 것을 볼수 있는데, 만약에 person, car 같은 특정한 class만 detection 할수 있는 방법이 있을까요?labels_to_names_seq 의 클래스 매핑에서 제외해버리면 되는 것이가요??
답변 1
1
권 철민
지식공유자
안녕하십니까
시각화 코드에 filtering 부분이 있습니다.
if score가 score_threshold(예:0.5) 보다 작을 경우 bbox를 시각화 하지 않습니다. 여기에 labels_to_names의 값이 person, car 인 경우만 시각화 하도록 추가해 주면 될 것 같습니다.
감사합니다.