preview_dataset.py 文件源码

def draw_spline(image, lane, ys, color=[0, 0, 255]):
    pts = [[x, ys[i]]for i, x in enumerate(lane) if not math.isnan(x)]
    if len(pts)-1 > 0:
        spline = interpolate.splrep([pt[1] for pt in pts], [pt[0] for pt in pts], k=len(pts)-1)
        for i in range(min([pt[1] for pt in pts]), max([pt[1] for pt in pts])):
            image[i, int(interpolate.splev(i, spline)), :] = color
    return image