From c8b4d04fb371e5fd115978bf2eb97a74f9a062d0 Mon Sep 17 00:00:00 2001
From: fangjiyuan <36502693+fangjiyuan@users.noreply.github.com>
Date: Thu, 16 Mar 2023 08:20:36 +0800
Subject: [PATCH] Update casia_hwdb.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

默认不用v2版函数会报错。
---
 dataset/casia_hwdb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dataset/casia_hwdb.py b/dataset/casia_hwdb.py
index 253c764..54b19a7 100755
--- 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