• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

mAP 계산하려면?

21.10.13 23:35 작성 조회수 458

0

선생님, 안녕하세요?
 
YOLO 결과를 받아서 mAP를 계산하려고 합니다.
그럴려면 강의에서 말씀하신 것처럼
모든 confidence에 대한 결과로 PR 그래프를 그려야 하니까
예측 결과를 구할 때
confidence_thres=0.001로 낮게 해야 하죠?
 
그리고 일반적인 예측시에는
confidence_thres=0.4처럼 정해서 원하는 정확도와 재현율을 찾구요.
 
그리고,
No mask와 Mask가 박스로 동시에 예측하는 경우가 있는데,
제가 알기로는 NMS에서 겹치는 박스시에는 제거하는 걸로 아는데,
그것은 동일 클래스인 경우에 해당되는 개념인가요?
 
제 생각이 맞는지 부연 설명 좀 부탁드립니다.
 
 
 

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

1. mAP 설명 드릴때 Confidence를 높은 것 부터 낮춰가면서 말씀 드린것은 이해를 돕고자 말씀을 드린것이고, 실제 mAP를 구현할 때는 예측된 Bounding box의  Confidence가 높은 순에서 낮은 순으로 모두 나열한 다음에 Ground truth와 겹치는 IOU가 0.5 이상인것을 찾아서 Precision, Recall 계산을 계속 해나가야 합니다. 

이게 실제 구현이 쉽지 않으실 것입니다. 참고하실만한 링크 먼저 확인해 보시기 바랍니다. 

https://github.com/Cartucho/mAP

2. NMS를 겹치는 동일 클래스로 할수도 있지만, 일반적으로는 클래스에 관계없이 겹치는 BOX에 대해서 제거 합니다. 

감사합니다.