1、图片维度增减
tf.expand_dims(input, 0) #(256, 256, 3) -> (1, 256, 256, 3)
input[tf.newaxis, ...] #(256, 256, 3) ->(1, 256, 256, 3)
1、图片维度增减
tf.expand_dims(input, 0) #(256, 256, 3) -> (1, 256, 256, 3)
input[tf.newaxis, ...] #(256, 256, 3) ->(1, 256, 256, 3)