preview_dataset.py 文件源码

python
阅读 19 收藏 0 点赞 0 评论 0

项目:ego-lane-analysis-system 作者: rodrigoberriel 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号