dearsky/hand-writing-recognition
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

update code and readme

Browse Source
This commit is contained in:
JiageWang
2019-08-24 21:27:08 +08:00
parent 2ffcef9ac7
commit b362e285a5
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
hwdb.py
View File

@@ -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))