explore.py 文件源码

def compare(test_id, angles, models):
    img = X_test[test_id]
    imgs = np.array([ndimage.rotate(img, rot, reshape=False) for rot in angles])

    all_probs = []
    all_matched = []
    for model in models:
        probs, matched = get_probs_matched(imgs, model, test_id)
        all_probs.append(probs)
        all_matched.append(matched)

    return all_probs, all_matched