-
Notifications
You must be signed in to change notification settings - Fork 1
/
makedataset.py
56 lines (46 loc) · 1.41 KB
/
makedataset.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# -*- coding: utf-8 -*-
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import os
import h5py
import sys
import numpy as np
import pandas as pd
import re
#subset_filepath=sys.argv[1]
subset_filepath = "C:/Users/kyeb2/PycharmProjects/project1/CASIA-HWDB1.1-cnn/src/HWDB1.1subset.hdf5"
with h5py.File(subset_filepath, 'r') as f:
trnx = []
for image in range(40000):
for pixel in (f['trn/x'][image][:]):
for row in range(64):
data = list()
for column in range(64):
data.append(int(pixel[row, column]))
# print(data)
trnx.append(data)
tstx=[]
for image in range(11946):
for pixel in (f['tst/x'][image][:]):
for row in range(64):
data = list()
for column in range(64):
data.append(int(pixel[row, column]))
# print(data)
tstx.append(data)
trnlabel=[]
for a in range(200):
for b in range(200):
trnlabel.append(b)
'''
tstlabel=[]
with open('C:/Users/kyeb2/PycharmProjects/project1/CASIA-HWDB1.1-cnn/src/tstlist.txt', 'w') as f:
for label in f:
tstlabel.append(f[label])
'''
def loadSimpData():
datMat = trnx
classLabels = trnlabel
return datMat,classLabels