TensorFlow数据集操作(使用slim和tfrecord)
在卷积神经网络中,输入的图像数据集都非常大,而且与其他数据不同,图像都需要以三维张量(height,width,channel)形式表示,这样使得神经网络读取数据非常麻烦。在TensorFlow框架中,有一种用的非常多的方法来处理数据集,就是tfrecord文件,它与TensorFlow的中层封装库slim搭配使用起来非常方便,下面我以DeepLabv3+中的数据处理代码为例解释一下该方法。该数...
原创
2019-01-21 22:38:20 ·
1494 阅读 ·
0 评论