tensorflow前处理 - 输入图片操作

1、图片维度增减

tf.expand_dims(input, 0)    #(256, 256, 3) -> (1, 256, 256, 3)
input[tf.newaxis, ...]　　#(256, 256, 3) ->(1, 256, 256, 3)

 

posted @ 2021-08-01 15:46  kuaqi  阅读(141)  评论(0)  编辑  收藏  举报