¿Podemos agregar el valor de confianza junto a cada etiqueta de bbox? ¿Dónde puedo encontrar cómo hacerlo?
Si desea generar confianza en la imagen de salida, debería poder modificar esta línea: https://github.com/thtrieu/darkflow/blob/22ac87a3cb1b585d62c5106aab2f203ead4b445f/darkflow/net/yolov2/test.py#L61
A lo siguiente:
cv2.putText(imgcv, mess + ("{0:.2f}".format(confidence)), (left, top - 12),
Avísame si eso funciona para ti. Si está utilizando YOLOv1, simplemente actualice la línea equivalente para test.py
en YOLOv1
@abagshaw Wow, es así de simple. Gracias, lo acabo de probar. Funciona genial.
También quiero que la carta sea más audaz y quiero poner un fondo detrás del texto como el darknet original. Pero intentaré averiguarlo yo mismo.
Gracias anw
Con esta nueva versión, el archivo que necesita editar no es test.py
sino predict.py
para nuevas personas que se pregunten.
Creo que esta debería ser una opción para el comando flow
, no solo algo para lo que el usuario necesita ajustar el código. Algo como --showConfidence
, supongo.
Comentario más útil
Si desea generar confianza en la imagen de salida, debería poder modificar esta línea: https://github.com/thtrieu/darkflow/blob/22ac87a3cb1b585d62c5106aab2f203ead4b445f/darkflow/net/yolov2/test.py#L61
A lo siguiente:
cv2.putText(imgcv, mess + ("{0:.2f}".format(confidence)), (left, top - 12),
Avísame si eso funciona para ti. Si está utilizando YOLOv1, simplemente actualice la línea equivalente para
test.py
en YOLOv1