如何從圖像的熱圖數據生成邊界框數據？

Question

我有一組圖像和一些單獨的熱圖數據（不完美地）解釋圖像的主題。熱圖數據與形狀（224,224,3）形成一個numpy陣列。我想從這個熱圖數據生成邊界框數據。

熱點地圖並不總是完美的，所以我想我想知道如果有人能想到一個聰明的方法來做到這一點。

這裏是我的熱圖數據應用到圖像會發生什麼情況的一些例子：

我發現在MATLAB解決，但我不知道如何閱讀這段代碼！不幸的是，我是一名Python程序員。 https://github.com/metalbubble/CAM/tree/master/bboxgenerator

任何人有任何想法如何處理這樣的事情？

Answer 1

我不太清楚你項目的熱圖數據是怎麼樣的，但在我看來你可以使用像Selective Search這樣的東西。你也可以看看interesting paper。也許你可以在你的數據集上使用這種方法。