使用多线程读取大量图片数据

程序在需要读取多幅图片时,
需要使用多线程来进行操作:

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord)

# use multithreading to output image_batch and label_batch
try:
   while not coord.should_stop() and i < 1:

   image_batch, label_batch = sess.run([image_batch, label_batch])

   # just test one batch
   for j in np.arange(FLAGS.train_batch_size):
   print('label: %d' % label_batch[j])
   # plt.imshow(image_batch[j, :, :, :])
   # plt.show()
   i += 1

except tf.errors.OutOfRangeError:
   print('done!')
finally:
   coord.request_stop()
   coord.join(threads)
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容