Я хочу посчитать объекты на изображении с тепловыми картами, сгенерированными нейронным net (U- Net). Я хочу достичь того, чего добился автор этой статьи: https://towardsdatascience.com/objects-counting-by-estimating-a-density-map-with-convolutional-neural-networks-c01086f3b3ec
Насколько я понял, я должен обучить нейронную сеть создавать тепловые карты, которые затем можно интегрировать в получить оценку количества объектов.
Я сражаюсь уже несколько дней, чтобы сгенерировать тренировочные данные для этого, но я не могу сгенерировать эти тепловые карты.
Есть Правильно, что мне нужны эти три строительных блока, или я неправильно здесь думаю?
1) Какой-то код, который делает KDE на основе некоторых координат / точек.
2) Какой-то код, который генерирует тепловую карту из этого KDE
3) Некоторый код, который затем интегрирует тепловую карту, так что на выходе получается число, близкое к счету координат
С помощью этого кода я смог бы затем комментировать некоторые изображения и генерировать из него обучающие данные, чтобы обучать сеть генерировать тепловые карты, которые затем можно интегрировать для получения количества объектов.
Я был бы очень благодарен за некоторую помощь или python код.