test_substance_to_tfrecord.py 文件源码

def prepare_packed(data, output_fname):
        print("prepare packed:")
        with tictoc():
            with tf.python_io.TFRecordWriter(output_fname) as writer:
                for atom_features in data:
                    value = np.packbits(atom_features.reshape(-1)).tobytes()
                    bit_features = tf.train.Feature(bytes_list = tf.train.BytesList(value=value))
                    features = {'bit_features' : bit_features}
                    example = tf.train.Example(features=tf.train.Features(feature=features))
                    serialized_example = example.SerializeToString()
                    writer.write(serialized_example)
        print("File size: {} bytes".format(os.path.getsize(output_fname)))