plot-submission.py 文件源码

def plot_prediction_overlay(subm_name, image_id, save_file=None):
    subm = pd.read_csv('subm/%s.csv.gz' % subm_name)
    img = load_image(image_id)

    if save_file:
        fig, ax = plt.subplots(figsize=(30, 30))
    else:
        fig, ax = plt.subplots()

    xmax = grid_sizes.loc[image_id, 'xmax']
    ymin = grid_sizes.loc[image_id, 'ymin']

    ax.imshow(np.rollaxis(img, 0, 3))

    # plotting, color by class type
    for cls in xrange(10):
        multi_poly = shapely.wkt.loads(subm.loc[(subm['ClassType'] == cls + 1) & (subm['ImageId'] == image_id), 'MultipolygonWKT'].values[0])

        #if cls == 3:
            #multi_poly = multi_poly.buffer(2e-5).buffer(-2e-5)

        for poly in multi_poly:
            coords = convert_geo_coords_to_raster(np.array(poly.exterior), img.shape[1:], (xmax, ymin))
            ax.add_patch(Polygon(coords, color=cls_colors[cls], lw=1.0, alpha=cls_alphas[cls]))

    plt.title(subm_name)

    if save_file:
        plt.savefig(save_file)
        plt.close()
    else:
        plt.show()