Update casia_hwdb.py

默认不用v2版函数会报错。
2023-03-16 08:20:36 +08:00
1 changed files with 1 additions and 1 deletions
--- a/dataset/casia_hwdb.py
+++ b/dataset/casia_hwdb.py
@@ -62,7 +62,7 @@ def parse_example_v2(record):
 def load_ds():
    input_files = ['dataset/HWDB1.1trn_gnt.tfrecord']
    ds = tf.data.TFRecordDataset(input_files)
-    ds = ds.map(parse_example)
+    ds = ds.map(parse_example_v2)
    return ds