tensorflow前处理

1、图片维度增减

tf.expand_dims(input, 0)    #(256, 256, 3) -> (1, 256, 256, 3)
input[tf.newaxis, ...]  #(256, 256, 3) ->(1, 256, 256, 3)
原文地址:https://www.cnblogs.com/wllwqdeai/p/15086827.html