Podemos adicionar o valor de confiança ao lado de cada rótulo bbox? Onde posso encontrar como fazer isso?
Se você deseja obter confiança na imagem de saída, deve ser capaz de modificar esta linha: https://github.com/thtrieu/darkflow/blob/22ac87a3cb1b585d62c5106aab2f203ead4b445f/darkflow/net/yolov2/test.py#L61
Para o seguinte:
cv2.putText(imgcv, mess + ("{0:.2f}".format(confidence)), (left, top - 12),
Deixe-me saber se isso funciona para você. Se você estiver usando YOLOv1, basta atualizar a linha equivalente para test.py
em YOLOv1
@abagshaw Uau, é tão simples. Obrigado, acabei de experimentar. Funciona bem.
Também quero que a letra seja mais ousada e quero colocar um fundo atrás do texto, como o darknet original. Mas vou tentar descobrir isso sozinho.
Obrigado anw
Com esta nova versão, o arquivo que você precisa editar não é test.py
mas sim predict.py
para novas pessoas que estão se perguntando.
Acho que esta deve ser uma opção para o comando flow
, não apenas algo para o qual o usuário precisa ajustar o código. Algo como --showConfidence
, eu acho.
Comentários muito úteis
Se você deseja obter confiança na imagem de saída, deve ser capaz de modificar esta linha: https://github.com/thtrieu/darkflow/blob/22ac87a3cb1b585d62c5106aab2f203ead4b445f/darkflow/net/yolov2/test.py#L61
Para o seguinte:
cv2.putText(imgcv, mess + ("{0:.2f}".format(confidence)), (left, top - 12),
Deixe-me saber se isso funciona para você. Se você estiver usando YOLOv1, basta atualizar a linha equivalente para
test.py
em YOLOv1