Как правильно создавать большие маски в Tensorflow? - PullRequest
0 голосов
/ 20 января 2020

Я получил карту тепла на промежуточном уровне моей сети.

mask = tf.cast (tf.less (threshold, heatmap), tf.float32)

Значение тензорной маски обычно выглядит так:

0.0 0.0 0.0 0.0 0.0 0.0
0.0 0.0 0.0 0.0 0.0 0.0
0.0 1.0 1.0 0.0 0.0 0.0
0.0 1.0 1.0 0.0 0.0 0.0
0.0 0.0 0.0 0.0 0.0 0.0
0.0 0.0 0.0 0.0 0.0 0.0

Теперь я надеюсь создать другую маску на основе вышеуказанного тензора с большей областью интересов:

0.0 0.0 0.0 0.0 0.0 0.0
0.5 0.5 0.5 0.5 0.0 0.0
0.5 1.0 1.0 0.5 0.0 0.0
0.5 1.0 1.0 0.5 0.0 0.0
0.5 0.5 0.5 0.5 0.0 0.0
0.0 0.0 0.0 0.0 0.0 0.0

Исходная область 1.0s не должны быть сжаты. Но я надеюсь, что ненулевые области станут немного больше вокруг исходной области, увеличив число 1 или добавив размытый край.

Будет ли это достигнуто какими-либо операциями Tensorflow?

...