예제 #1
0
파일: icbc.py 프로젝트: zhxgigi/toolkids
def classify_singlechar():
    singlechardir =  r'D:\projects\python\captchabreak\data\icbc\icbc1213_output'
    outputdir = r'D:\projects\python\captchabreak\data\icbc\icbc_c_output'
    try:
        os.mkdir(outputdir)
    except:
        pass
    
    traindata = r'D:\projects\python\captchabreak\data\icbc\icbcaudit'
    cc = Classifier(traindata, class_sample_num=20, size = 20)
    cc.train()
    count =0
    for subdir in cc.classify_map:
        try:
            os.mkdir(os.path.join(outputdir, subdir))
        except:
            pass
    for root, dirs, files in os.walk(singlechardir):
        for file in files:
            path = os.path.join(root, file)
            img = cv.LoadImage(path, 0)
            response = cc.classify_single_char(img)
            
            outputfile = os.path.join(outputdir, response, 'icbc_%d.jpg'%count)
            cv.SaveImage(outputfile, img)
            count += 1
예제 #2
0
파일: douban.py 프로젝트: zhxgigi/toolkids
def main():
    traindata = r'D:\projects\python\captchabreak\data\doubanaudit'
    douban = Classifier(traindata, 5)
    douban.train()
    
    testdata = r'D:\projects\python\captchabreak\data\doubanseg'
    outputdir = r'D:\projects\python\captchabreak\data\doubanoutput'
    count =0
    for subdir in douban.classify_map:
        try:
            os.mkdir(os.path.join(outputdir, subdir))
        except:
            pass
    for root, dirs, files in os.walk(testdata):
        for file in files:
            path = os.path.join(root, file)
            img = cv.LoadImage(path, 0)
            response = douban.classify_single_char(img)
            
            outputfile = os.path.join(outputdir, response, 'douban_%d.jpg'%count)
            cv.SaveImage(outputfile, img)
            count += 1