• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

detection 할 class 관해 질문이 있습니다.

21.07.13 17:34 작성 조회수 90

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 인 경우만 시각화 하도록 추가해 주면 될 것 같습니다. 

감사합니다.