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
preview_dataset.py 文件源码
python
阅读 19
收藏 0
点赞 0
评论 0
评论列表
文章目录