cosense3d.agents.viewer package

Subpackages

Submodules

cosense3d.agents.viewer.gl_viewer module

class cosense3d.agents.viewer.gl_viewer.GLViewer(name: str, parent=None)[source]

Bases: GLViewWidget

addBox()[source]
box()[source]
change_visibility(key, visible)[source]
drawRectangle()[source]
draw_axes()[source]
draw_depth_buffer()[source]
evt_pos_to_world(pos1, pos2=None)[source]
Args:

pos1: center pos if pos2 is None else start post of a region pos2: end pos of a region

get_point_depth(x, y)[source]
get_region_depth(p1: QPoint, p2: QPoint) → ndarray[source]
Args:

p1: start point of region. p2: end point of region

highlightBox(pos)[source]
initializeGL()[source]

Initialize items that were not initialized during addItem().

keyPressEvent(self, a0: QKeyEvent | None)[source]
keyReleaseEvent(self, a0: QKeyEvent | None)[source]
model_pose_to_world(x, y, z)[source]
mouseDoubleClickEvent(self, a0: QMouseEvent | None)[source]
mouseMoveEvent(self, a0: QMouseEvent | None)[source]
mousePressEvent(self, a0: QMouseEvent | None)[source]
mouseReleaseEvent(self, a0: QMouseEvent | None)[source]
paintGL(region=None, viewport=None, useItemNames=False)[source]

viewport specifies the arguments to glViewport. If None, then we use self.opts[‘viewport’] region specifies the sub-region of self.opts[‘viewport’] that should be rendered. Note that we may use viewport != self.opts[‘viewport’] when exporting.

paintRect()[source]
refresh(data_dict, visible_keys=['globalGT'], color_mode='united', **kwargs)[source]
removeActivate()[source]
removeHeilight()[source]
removeRectangle()[source]
selectHeilight()[source]
updateFrameData(pcds, local_label=None, global_label=None, local_det=None, global_det=None, predecessor=None, successor=None, successor_gt=None, pcd_color='united')[source]
updateLabel(local_labels, global_labels, local_det, global_det, successor=None, successor_gt=None, predecessor=None)[source]
updatePCDs(pcds, color_mode='united', **kwargs)[source]

cosense3d.agents.viewer.img_anno3d_viewer module

class cosense3d.agents.viewer.img_anno3d_viewer.ImgAnno3DViewer(dpi=50)[source]

Bases: FigureCanvasQTAgg

refresh(data, **kwargs)[source]

cosense3d.agents.viewer.img_viewer module

class cosense3d.agents.viewer.img_viewer.ImgViewer(dpi=100, mean=None, std=None)[source]

Bases: FigureCanvasQTAgg

refresh(data, **kwargs)[source]

cosense3d.agents.viewer.output_viewer module

class cosense3d.agents.viewer.output_viewer.BEVDenseCanvas(lidar_range=None, **kwargs)[source]

Bases: MplCanvas

refresh(data, **kwargs)[source]
class cosense3d.agents.viewer.output_viewer.BEVSparseCanvas(lidar_range=None, s=4, **kwargs)[source]

Bases: MplCanvas

refresh(data, **kwargs)[source]
class cosense3d.agents.viewer.output_viewer.DetectionCanvas(lidar_range=None, topk_ctr=0, **kwargs)[source]

Bases: MplCanvas

refresh(data, **kwargs)[source]
class cosense3d.agents.viewer.output_viewer.DetectionScoreMap(lidar_range=None, s=4, **kwargs)[source]

Bases: MplCanvas

refresh(data, **kwargs)[source]
class cosense3d.agents.viewer.output_viewer.MplCanvas(data_keys, width=5, height=4, dpi=100, title='plot', nrows=1, ncols=1)[source]

Bases: FigureCanvasQTAgg

update_title(scenario, frame, cav_id)[source]
class cosense3d.agents.viewer.output_viewer.OutputViewer(plots, parent=None)[source]

Bases: QWidget

refresh(data, **kwargs)[source]
class cosense3d.agents.viewer.output_viewer.SparseDetectionCanvas(lidar_range=None, topk_ctr=0, **kwargs)[source]

Bases: MplCanvas

refresh(data, **kwargs)[source]

cosense3d.agents.viewer.utils module

cosense3d.agents.viewer.utils.circular_mask(arr_length, center, radius)[source]
cosense3d.agents.viewer.utils.depth_min(depths, center, r=10) → float[source]

Module contents