1.nms
(1) proceso nms
a. Primero, eliminamos todos los cuadros de predicción con puntuaciones de confianza inferiores al umbral de confianza a.
b. Luego, en la misma imagen, ordenamos todas las casillas pronosticadas según el nivel de confianza de mayor a menor según la categoría (excepto la categoría de fondo, porque la categoría de fondo no necesita ser NMS), y tomamos la casilla con la confianza más alta como la que queremos mantener La primera caja predicha de esta categoría,
c. Luego calcule el IoU de las casillas de predicción restantes y en orden,
d. Eliminar los cuadros de predicción cuyo IoU es mayor que el umbral b de IoU (de hecho, en la implementación del código, las puntuaciones de confianza de estos cuadros de predicción que se eliminarán se establecen en 0),
f) Al final de la primera iteración, hemos eliminado las casillas con alta coincidencia con la primera casilla.
(2) umbral nms
Propósito de nms: filtrar cajas redundantes
Entonces, el umbral de iou aquí es el iou que controla el marco redundante y el marco de predicción con la mayor confianza. Cuanto más pequeño es el iou aquí, más estricto es.
2. Calcula el mapa
(1) Proceso de cálculo
a. Primero recorra el objeto de verdad del suelo en la imagen, luego extraiga los objetos gt de una categoría determinada que queremos calcular, y luego lea el marco de detección de esta categoría detectado por el detector (no se preocupe por otras categorías ) ,
b. Luego filtre la casilla cuyo puntaje de confianza sea inferior al umbral de confianza (es decir, el llamado umbral de confianza c),
c. Ordene los marcos de detección restantes según el puntaje de confianza de mayor a menor y primero determine si la IoU entre el marco de detección con el puntaje de confianza más alto y el gt bbox es mayor que el umbral de IoU (es decir, el IoU mencionado anteriormente). Umbral de IoU d), si Si el IoU es mayor que el umbral de IoU establecido, se considera TP.
d. Marque este gt_bbox como detectado (los marcos de detección redundantes subsiguientes del mismo GT se consideran todos como FP, por lo que primero se ordena según el puntaje de confianza de mayor a menor, y el marco de detección con el puntaje de confianza más alto es el primero en compararse con la comparación del umbral de IoU, si es mayor que el umbral de IoU, se considera como TP, y los marcos de detección posteriores del mismo objeto gt se consideran como FP) ,
f. Si el IoU es menor que el umbral, se planifica directamente en el FP.
(2) El umbral del pagaré del mapa
El umbral del IOU aquí es para controlar el IOU entre el cuadro de predicción con la confianza más alta y el valor real. Cuanto más pequeño es el IOU aquí, más grande es el mapa calculado.