def data_generator(data_dir):
reader = tf.TextLineReader()
queue = tf.train.string_input_producer([data_dir])
_, value = reader.read(queue)
coord = tf.train.Coordinator()
sess = tf.Session()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)
while True:
v = sess.run(value)
[data, label] = v.split(b"|")
data = np.array(json.loads(data.decode("utf-8")))
label = np.array(json.loads(label.decode("utf-8")))
yield (data, label)
coord.request_stop()
coord.join(threads)
sess.close()
评论列表
文章目录