Documentation

Yolo3D

Input

loc:
  • Required: yes
  • Type: float16
  • Description: coordinates of the object to be detected
  • Restriction: NCHW; supported dimension: [1,64,36,90]
obj_pred:
  • Required: yes
  • Type: float16
  • Description: class confidence of the bounding box
  • Restriction: NCHW; supported dimension: [1,16,36,90]
cls0_pred:
  • Required: yes
  • Type: float16
  • Description: class confidence
  • Restriction: NCHW; supported dimension: [1,128,36,90]
argmax_index:
  • Required: yes
  • Type: int32
  • Description: index of the maximum value
  • Restriction: NCHW; supported dimension: [1,16,36,90]
ori_pred:
  • Required: yes
  • Type: float16
  • Description: prediction of the rotation angle
  • Restriction: NCHW; supported dimension: [1,16,36,90]
dim:
  • Required: yes
  • Type: float16
  • Description: size of the bounding box
  • Restriction: NCHW; supported dimension: [1,48,36,90]
brvis_filer:
  • Required: yes
  • Type: float16
  • Description: input of the brake light visibility
  • Restriction: NCHW; supported dimension: [1,16,36,90]
ltvis_filer:
  • Required: yes
  • Type: float16
  • Description: input of the left turn light visibility
  • Restriction: NCHW; supported dimension: [1,16,36,90]
rtvis_filer:
  • Required: yes
  • Type: float16
  • Description: input of the right turn light visibility
  • Restriction: NCHW; supported dimension: [1,16,36,90]
brswt_filer:
  • Required: yes
  • Type: float16
  • Description: confidence for brake light on
  • Restriction: NCHW; supported dimension: [1,16,36,90]
ltswt_filer:
  • Required: yes
  • Type: float16
  • Description: confidence for left turn light on
  • Restriction: NCHW; supported dimension: [1,16,36,90]
rtswt_filer:
  • Required: yes
  • Type: float16
  • Description: confidence for right turn light on
  • Restriction: NCHW; supported dimension: [1,16,36,90]
vis:
  • Required: yes
  • Type: float16
  • Description: visibility of the obstacle
  • Restriction: NCHW; supported dimension: [1,64,36,90]
area:
  • Required: yes
  • Type: float16
  • Description: area occupied by the obstacle
  • Restriction: NCHW; supported dimension: [1,64,36,90]
cut:
  • Required: yes
  • Type: float16
  • Description: cut-in scenario detection
  • Restriction: NCHW; supported dimension: [1,64,36,90]
conti_dim0:
  • Required: yes
  • Type: float16
  • Description: x coordinate of the feature map
  • Restriction: NCHW; supported dimension: [1,16,36,90]
conti_dim1:
  • Required: yes
  • Type: float16
  • Description: y coordinate of the feature map
  • Restriction: NCHW; supported dimension: [1,16,36,90]
anchor_dim0:
  • Required: yes
  • Type: float16
  • Description: anchor width
  • Restriction: NCHW; supported dimension: [1,16,36,90]
anchor_dim1:
  • Required: yes
  • Type: float16
  • Description: anchor height
  • Restriction: NCHW; supported dimension: [1,16,36,90]
expand_data:
  • Required: yes
  • Type: float16
  • Description: extended data
  • Restriction: supported dimension [8,128]

Attribute

num_classes:
  • Required: yes
  • Type: int64
  • Description: number of classes
num_areas:
  • Required: yes
  • Type: int64
  • Description: number of areas
with_box3d:
  • Required: yes
  • Type: boolean
  • Description: whether 3D bounding boxes are included
with_lights:
  • Required: yes
  • Type: boolean
  • Description: whether vehicle light information is included
with_ratios:
  • Required: yes
  • Type: boolean
  • Description: whether to adjust the ratio
max_exp_power:
  • Required: yes
  • Type: float32
  • Description: maximum value range of exp during coordinate calculation
min_exp_power:
  • Required: yes
  • Type: float32
  • Description: minimum value range of exp during coordinate calculation

Output

output_bbox:
  • Required: yes
  • Type: float16
  • Description: output bounding box
  • Restriction: NCHW; supported dimension: [1,32,576,90]
output_score:
  • Required: yes
  • Type: float16
  • Description: confidence of the output bounding box
  • Restriction: NCHW; supported dimension: [1,8,576,90]