vis4d.op.box.encoder.qd_3dt

3D bounding box coder.

Classes

QD3DTBox3DDecoder([center_scale, ...])	3D bounding box decoder based on qd_3dt.
QD3DTBox3DEncoder([center_scale, ...])	3D bounding box encoder based on qd_3dt.

class QD3DTBox3DDecoder(center_scale=10.0, depth_log_scale=2.0, dim_log_scale=2.0, num_rotation_bins=2)

3D bounding box decoder based on qd_3dt.

Creates an instance of the class.

__call__(boxes_2d, boxes_deltas, intrinsics)

Decode the predicted boxes_deltas according to given 2D boxes.

Return type:

Tensor

class QD3DTBox3DEncoder(center_scale=10.0, depth_log_scale=2.0, dim_log_scale=2.0, num_rotation_bins=2, bin_overlap=0.16666666666666666)

3D bounding box encoder based on qd_3dt.

Init.

__call__(boxes, boxes3d, intrinsics)

Encode deltas between 2D boxes and 3D boxes given intrinsics.

Return type:

Tensor