predict.py 文件源码

def predict_knn(image_file):
    image = cv2.imdecode(np.fromstring(image_file.read(), np.uint8), cv2.CV_LOAD_IMAGE_UNCHANGED)
    if image is not None:
        features = np.array([extract_color_histogram(image)])
        loaded_model = pickle.load(open(MODEL_PATH + "/knn_model.sav", 'rb'))

        return loaded_model.predict(features)[0]
    else:
        raise "Failed"