From b362e285a59852ce2a5942b456f1cfbe63e11ab1 Mon Sep 17 00:00:00 2001
From: JiageWang <1076050774@qq.com>
Date: Sat, 24 Aug 2019 21:27:08 +0800
Subject: [PATCH] update code and readme

---
 hwdb.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/hwdb.py b/hwdb.py
index b3a8abb..448886e 100644
--- a/hwdb.py
+++ b/hwdb.py
@@ -37,12 +37,12 @@ if __name__ == '__main__':
     dataset = HWDB(transform=transform, path=r'C:\Users\Administrator\Desktop\hand-writing-recognition\data')
     print(dataset.train_size)
     print(dataset.test_size)
-    # for i in [1020, 120, 2000, 6000, 1000]:
-    #     img, label = dataset.get_sample(i)
-    #     img = img[0]
-    #     print(label)
-    #     plt.imshow(img, cmap='gray')
-    #     plt.show()
+    for i in [1020, 120, 2000, 6000, 1000]:
+        img, label = dataset.get_sample(i)
+        img = img[0]
+        print(label)
+        plt.imshow(img, cmap='gray')
+        plt.show()
 
     train_loader, test_loader = dataset.get_loader()
     print(len(train_loader))