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!
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
Tags und Links tensorflow conv-neural-network