add

2019-06-05 23:49:20 +08:00
parent 919d89af4b
commit 0d9ea44929
3 changed files with 136 additions and 1 deletions
--- a/dataset/casia_hwdb.py
+++ b/dataset/casia_hwdb.py
@@ -57,7 +57,7 @@ def parse_example(record):


 def load_ds():
-    input_files = ['casia_hwdb_1.0_1.1.tfrecord']
+    input_files = ['dataset/hwdb_11.tfrecord']
    ds = tf.data.TFRecordDataset(input_files)
    ds = ds.map(parse_example)
    return ds