导出数据集以驱动colab

时间:2019-01-12 00:32:54

标签: python-3.x tensorflow deep-learning google-colaboratory

我的机器导出中有此代码 两个fils train.tfrecord和eval.tfrecord。 但是当我将其上传到Colab时,不会导出任何内容

写数据集

train_writer = tf.python_io.TFRecordWriter(FLAGS.train_output_path)
eval_writer = tf.python_io.TFRecordWriter(FLAGS.eval_output_path)

process_dataset(
    train_writer=train_writer,
    eval_writer=eval_writer,
    data_dir=FLAGS.data_dir)

train_writer.close() eval_writer.close()
     

Github的完整代码:   https://github.com/frankgu/3d-DenseNet/blob/sagemaker/prepare_data_main.py

协作NoteBook

https://colab.research.google.com/drive/1u9Rf9l6Sp9GZEpgmVAa8v29ZcbGrVf9d

0 个答案:

没有答案