identify_code.py 文件源码

def _get_data_(num, pic_dict=path):
    # ???????
    imgdata = []
    labeldata = []
    # ?????
    checkcount = 50
    nowcount = 0

    # ???????????
    class Getoutofloop(Exception):
        pass

    try:
        while True:
            for root, dirs, files in os.walk(pic_dict):
                for dir in dirs:
                    for img in os.walk(os.path.join(root, dir)):
                        for imagename in img[2]:
                            if random.randint(0, 80) < 2:
                                image = cv2.imread(str(img[0]) + '/' + str(imagename), cv2.IMREAD_GRAYSCALE)
                                image = image.astype(np.float32)
                                image = np.multiply(image, 1.0 / 255.0)
                                imgdata.append(np.ravel(image))
                                tmplabel = img[0]
                                tmplabel = tmplabel[len(tmplabel) - 1]
                                labeldata.append(tmplabel)
                                nowcount += 1
                                if nowcount == checkcount:
                                    raise Getoutofloop()
                            else:
                                continue
    except Getoutofloop:
        pass
    imgdata = np.array(imgdata)
    labeldata = dc._one_hot_(np.array(labeldata))
    return imgdata, labeldata


# TensorFlow????