tensorflow中 tf.train.slice_input_producer() 函数和 tf.train.batch() 函数
原创:https://blog.csdn.net/dcrmg/article/details/79776876别人总结的转载方便自己以后看tensorflow数据读取机制tensorflow中为了充分利用GPU,减少GPU等待数据的空闲时间,使用了两个线程分别执行数据读入和数据计算。具体来说就是使用一个线程源源不断的将硬盘中的图片数据读入到一个内存队列中,另一个线程负责计算任务,所需数据直接从内存...
转载
2018-05-18 16:50:19 ·
1129 阅读 ·
0 评论