Преобразование ограничивающих рамок прогноза в тепловую карту - PullRequest
0 голосов
/ 01 апреля 2020

У меня есть модель обнаружения объекта, которая предсказывает один класс и возвращает набор ограничивающих рамок с различными баллами. Я хочу преобразовать эти ограничивающие рамки в тепловую карту на основе результатов и наложить ее поверх входного изображения. Таким образом, область, где объект видим, сильно выделена.

Предсказанные ограничивающие рамки наверняка будут перекрываться в некоторых пикселях; поэтому я думал об агрегировании баллов по этим пикселям. Я также слышал о ядрах плотности, которые могут помочь сгладить вероятности.

Я не могу найти способ собрать все это, используя Python с PIL или matplotlib. Есть мысли, пожалуйста?

...