Tensorflow png