def __init__(self):
collect_dir = 'captcha/collect'
label = []
train_file = []
for i in os.listdir(collect_dir):
for y in os.listdir(collect_dir + '/' + i):
#print i
label.append(ord(i))
#print y
train_file.append(collect_dir + '/' + i + '/' + y)
train_data = [cv2.imread(i, 0) for i in train_file]
train = np.array(train_data).reshape(-1, 400).astype(np.float32)
label = np.array(label).reshape(-1)
self.knn = cv2.KNearest()
self.knn.train(train, label)
newknn.py 文件源码
python
阅读 21
收藏 0
点赞 0
评论 0
评论列表
文章目录