Upsampling-Feature-Maps in TensorFlow

8

Ich möchte ein Faltungs-Dekonvolutionsnetzwerk für ein Bildsegmentierungsprojekt implementieren. Im Dekonvolutionsteil plane ich, die Merkmalskarte um 2 zu erhöhen, z. Das Original-Feature-Map hat die Dimension 64 * 64 * 4 und ich möchte es in 128 * 128 * 4 upsamplen. Kennt jemand eine Tensor-Operation, die das tut? Danke!

    
Wang Duo 11.04.2016, 11:36
quelle

1 Antwort

7

Sie könnten tf.image.resize_images() verwenden. Es verwendet Stapel von Bildern oder Einzelbildern und unterstützt die gängigsten Methoden wie bilinear und nearest_neighbor .

Hier ist der Link zur TensorFlow-API-Referenz: Größenanpassung

Sie können auch sehen, wie die Upsampling-Operation in einer übergeordneten API wie tflearn implementiert wird. Sie können upsample_2d und upscore_layer in ihrem Github Repo finden: conv.py

Hinweis: In älteren TF-Versionen könnte die Ausgabe in tf.float32 umgewandelt werden

    
chillinger 27.07.2016 20:04
quelle