Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 2.16 KB
Newer Older
zhangshilong's avatar
refactor directory  
zhangshilong committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38 
# Copyright (c) OpenMMLab. All rights reserved.
# yapf:disable
from .box_np_ops import (box2d_to_corner_jit, box3d_to_bbox,
                         box_camera_to_lidar, boxes3d_to_corners3d_lidar,
                         camera_to_lidar, center_to_corner_box2d,
                         center_to_corner_box3d, center_to_minmax_2d,
                         corner_to_standup_nd_jit, corner_to_surfaces_3d,
                         corner_to_surfaces_3d_jit, corners_nd,
                         create_anchors_3d_range, depth_to_lidar_points,
                         depth_to_points, get_frustum, iou_jit,
                         minmax_to_corner_2d, points_in_convex_polygon_3d_jit,
                         points_in_convex_polygon_jit, points_in_rbbox,
                         projection_matrix_to_CRT_kitti, rbbox2d_to_near_bbox,
                         remove_outside_points, rotation_points_single_angle,
                         surface_equ_3d)
# yapf:enable
from .iou3d_calculator import (AxisAlignedBboxOverlaps3D, BboxOverlaps3D,
                               BboxOverlapsNearest3D,
                               axis_aligned_bbox_overlaps_3d, bbox_overlaps_3d,
                               bbox_overlaps_nearest_3d)
from .transforms import bbox3d2result, bbox3d2roi, bbox3d_mapping_back

__all__ = [
    'box2d_to_corner_jit', 'box3d_to_bbox', 'box_camera_to_lidar',
    'boxes3d_to_corners3d_lidar', 'camera_to_lidar', 'center_to_corner_box2d',
    'center_to_corner_box3d', 'center_to_minmax_2d',
    'corner_to_standup_nd_jit', 'corner_to_surfaces_3d',
    'corner_to_surfaces_3d_jit', 'corners_nd', 'create_anchors_3d_range',
    'depth_to_lidar_points', 'depth_to_points', 'get_frustum', 'iou_jit',
    'minmax_to_corner_2d', 'points_in_convex_polygon_3d_jit',
    'points_in_convex_polygon_jit', 'points_in_rbbox',
    'projection_matrix_to_CRT_kitti', 'rbbox2d_to_near_bbox',
    'remove_outside_points', 'rotation_points_single_angle', 'surface_equ_3d',
    'BboxOverlapsNearest3D', 'BboxOverlaps3D', 'bbox_overlaps_nearest_3d',
    'bbox_overlaps_3d', 'AxisAlignedBboxOverlaps3D',
    'axis_aligned_bbox_overlaps_3d', 'bbox3d_mapping_back', 'bbox3d2roi',
    'bbox3d2result'
]