昇腾社区首页
EN
注册

Yolo3D

输入

loc:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:检测目标的坐标
  • 规格限制:NCHW,支持维度[1,64,36,90]
obj_pred:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:检测框的类别置信度
  • 规格限制:NCHW,支持维度[1,16,36,90]
cls0_pred:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:类别置信度
  • 规格限制:NCHW,支持维度[1,128,36,90]
argmax_index:
  • 是否必填:必填
  • 数据类型:int32
  • 参数解释:最大值的下标
  • 规格限制:NCHW,支持维度[1,16,36,90]
ori_pred:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:旋转角的预测
  • 规格限制:NCHW,支持维度[1,16,36,90]
dim:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:检测框的大小
  • 规格限制:NCHW,支持维度[1,48,36,90]
brvis_filer:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:刹车灯可见度的输入
  • 规格限制:NCHW,支持维度[1,16,36,90]
ltvis_filer:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:左转向灯的可见度输入
  • 规格限制:NCHW,支持维度[1,16,36,90]
rtvis_filer:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:右转向灯的可见度输入
  • 规格限制:NCHW,支持维度[1,16,36,90]
brswt_filer:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:刹车灯亮起的置信度
  • 规格限制:NCHW,支持维度[1,16,36,90]
ltswt_filer:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:左转向灯亮起的置信度
  • 规格限制:NCHW,支持维度[1,16,36,90]
rtswt_filer:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:右转向灯亮起的置信度
  • 规格限制:NCHW,支持维度[1,16,36,90]
vis:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:障碍物的可见度
  • 规格限制:NCHW,支持维度[1,64,36,90]
area:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:障碍物所占的区域
  • 规格限制:NCHW,支持维度[1,64,36,90]
cut:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:cut-in场景检测
  • 规格限制:NCHW,支持维度[1,64,36,90]
conti_dim0:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:feature map的x下标
  • 规格限制:NCHW,支持维度[1,16,36,90]
conti_dim1:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:feature map的y下标
  • 规格限制:NCHW,支持维度[1,16,36,90]
anchor_dim0:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:anchor的W
  • 规格限制:NCHW,支持维度[1,16,36,90]
anchor_dim1:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:anchor的H
  • 规格限制:NCHW,支持维度[1,16,36,90]
expand_data:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:扩展的数据
  • 规格限制:支持维度[8,128]

属性

num_classes:
  • 是否必填:必填
  • 数据类型:int64
  • 参数解释:类别的个数
num_areas:
  • 是否必填:必填
  • 数据类型:int64
  • 参数解释:area的个数
with_box3d:
  • 是否必填:必填
  • 数据类型:boolean
  • 参数解释:是否包含3D检测框
with_lights:
  • 是否必填:必填
  • 数据类型:boolean
  • 参数解释:是否包含车灯信息
with_ratios:
  • 是否必填:必填
  • 数据类型:boolean
  • 参数解释:是否调节比例
max_exp_power:
  • 是否必填:必填
  • 数据类型:float32
  • 参数解释:计算坐标时exp的最大取值范围
min_exp_power:
  • 是否必填:必填
  • 数据类型:float32
  • 参数解释:计算坐标时exp的最小取值范围

输出

output_bbox:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:输出的检测框
  • 规格限制:NCHW,支持维度[1,32,576,90]
output_score:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:输出框的置信度
  • 规格限制:NCHW,支持维度[1,8,576,90]