newknn.py 文件源码

python
阅读 21 收藏 0 点赞 0 评论 0

项目:ustc-grade-automatic-notification 作者: zzh1996 项目源码 文件源码
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)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号