utils.Utils_label

Classes

UtilsLabel 

UtilsLabel()

Attributes

occ_point_max instance-attribute 
occ_point_max = config.common_config.occ_point_max
occlusion_th instance-attribute 
occlusion_th = config.common_config.occlusion_th

Functions

get_care_labels staticmethod 
get_care_labels(bg_labels, care_key = None) -> Tuple[List[Any], List[Any]]
Parameters:
  • bg_labels –

    背景标签

  • care_key –

    关注物体类别
Returns:
  • Tuple[List[Any], List[Any]] –

    关注的物体标签，不关注的物体标签
get_image_occlusion_ratio 
get_image_occlusion_ratio(box1: numpy.ndarray, boxes2: numpy.ndarray) -> float
Parameters:
  • box1 (ndarray) –

    当前物体在图片中的2d box

  • boxes2 (ndarray) –

    其余物体在图片中的2d box
Returns:
  • float –

    当前物体被遮挡的比率
get_labels staticmethod 
get_labels(rz_degree: float, lidar_box: o3d.geometry.OrientedBoundingBox, calib_info: dict, image_box: List[float], truncation_ratio: float) -> List[str]
get_occlusion_level 
get_occlusion_level(occlusion_ratio: float) -> int
Parameters:
  • occlusion_ratio (float) –

    物体被遮挡的比率
Returns:
  • int –

    物体occlusion标签值
get_truncation_ratio staticmethod 
get_truncation_ratio(box1: List[float], box2: List[float]) -> float
sort_labels staticmethod 
sort_labels(labels_input)
update_bg_init_label 
update_bg_init_label(labels, init_pc, combine_pc, corners)
update_obj_inserted_labels 
update_obj_inserted_labels(labels, infos, delete_points_mask)
update_occ_only_image 
update_occ_only_image(labels: numpy.ndarray) -> list
Parameters:
  • labels (ndarray) –

    物体标签
Returns:
  • list –

    更新occlusion后的物体标签
update_single_init_label 
update_single_init_label(label, init_pc, combine_pc, corner)
update_single_labels 
update_single_labels(label, info, delete_points_mask)