Chengfang Lu
e72bc061c5
Major changes: - New frontend (platform/web/): Vite + React 18 + TypeScript + Tailwind - 4-module navigation: 数据送标 / 模型管理 / 车队管理 / 系统管理 - Data catalog with charts (DMS/ADAS/Lane 3-tab view) - Quality review workflow (标注质检): Good/Fine/Bad scoring with auto-advance - Audit enhancements: batch operations, rejection categories, Feishu notifications - Operation audit log (操作日志) - World model simulation studio (仿真工坊) - Dataset version management with snapshots and diff - ADAS 7-class dataset integration (138K images organized + compressed) - User management with Feishu integration and pagination - CRUD/search/filter on all pages, card layout redesign - PIL-optimized image overlay rendering - Auto-snapshot on build, in_review workflow stage - Removed embedded algorithm code (now in workspace)
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from scipy.interpolate import InterpolatedUnivariateSpline
|
|
import numpy as np
|
|
|
|
|
|
class Lane:
|
|
def __init__(self, points=None, invalid_value=-2., metadata=None):
|
|
super(Lane, self).__init__()
|
|
self.curr_iter = 0
|
|
self.points = points
|
|
self.invalid_value = invalid_value
|
|
self.function = InterpolatedUnivariateSpline(points[:, 1],
|
|
points[:, 0],
|
|
k=min(3,
|
|
len(points) - 1))
|
|
self.min_y = points[:, 1].min() - 0.01
|
|
self.max_y = points[:, 1].max() + 0.01
|
|
|
|
self.metadata = metadata or {}
|
|
|
|
def __repr__(self):
|
|
return '[Lane]\n' + str(self.points) + '\n[/Lane]'
|
|
|
|
def __call__(self, lane_ys):
|
|
lane_xs = self.function(lane_ys)
|
|
|
|
lane_xs[(lane_ys < self.min_y) |
|
|
(lane_ys > self.max_y)] = self.invalid_value
|
|
return lane_xs
|
|
|
|
def to_array(self, cfg):
|
|
sample_y = cfg.sample_y
|
|
img_w, img_h = cfg.ori_img_w, cfg.ori_img_h
|
|
ys = np.array(sample_y) / float(img_h)
|
|
xs = self(ys)
|
|
valid_mask = (xs >= 0) & (xs < 1)
|
|
lane_xs = xs[valid_mask] * img_w
|
|
lane_ys = ys[valid_mask] * img_h
|
|
lane = np.concatenate((lane_xs.reshape(-1, 1), lane_ys.reshape(-1, 1)),
|
|
axis=1)
|
|
return lane
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def __next__(self):
|
|
if self.curr_iter < len(self.points):
|
|
self.curr_iter += 1
|
|
return self.points[self.curr_iter - 1]
|
|
self.curr_iter = 0
|
|
raise StopIteration
|