trainingandtesting.py 文件源码

def testClassifier(foldername,classifier):
    clf = joblib.load(classifier)
    os.chdir(foldername)
    correct = 0
    total = 0
    for filename in glob.iglob('*.png'):
        img = cv2.imread(filename,-1)
        roi = hog(img.reshape((27, 35)), orientations=9, pixels_per_cell=(9, 7), cells_per_block=(1, 1), visualise=False)
        preditcion = clf.predict(roi)
        if preditcion == filename[0]:
            correct += 1
        total += 1
    print(total)
    print(correct)