Как запретить тензорному потоку показывать оценку обнаружения для каждого класса, чтобы сделать предсказание ограничивающей рамки читаемым

Я обучил модель тензорного потока с помощью API обнаружения объектов Tensorflow для обнаружения чисел в тексте. Однако я не вижу рамок, окружающих цифры, поскольку оценки обнаружения и класс, которые печатаются на изображении, делают его нечитаемым. Мой текущий результат для изображения выглядит примерно так: predicted_output

Может ли кто-нибудь помочь мне получить четкую ограничивающую рамку вокруг изображения, чтобы я мог видеть, правильно ли обучена модель.

Я пробовал изменить шрифт и толщину линии в visualizations_util файле в object_detection/utils, но это не работает


person Jitesh Malipeddi    schedule 17.07.2019    source источник
comment
Не могли бы вы дать ссылку на блокнот на GitHub или другой платформе, чтобы мы могли попробовать код?   -  person nsidn98    schedule 18.07.2019


Ответы (1)


Я действительно понял это. Мне пришлось перейти к функции visualize_boxes_and_labels_on_image_array в файле visulaizations_utils.py file и установить значение skip_scores на False

person Jitesh Malipeddi    schedule 19.07.2019