Python Dataset.get_test_data示例

示例#1

def evaluate(model_path, batch_size=128):
    print("Evaluating model {}".format(model_path))
    model = load_model(model_path)
    ds = Dataset()
    X, y = ds.get_test_data()

    gen = DataLoader(X, y, IMG_WIDTH, IMG_HEIGHT, batch_size)
    metrics = model.evaluate_generator(gen, steps=len(gen), workers=8)
    pprint(dict(zip(model.metrics_names, metrics)))