↓ 1 callersFunctionmask_boxes_outside_range_numpy Parameters ---------- boxes: np.ndarray (N, 7) [x, y, z, dx, dy, dz, heading], (x, y, z) is the box center limit_range: list
v2xvit/utils/box_utils.py:360
Method__init__(self, dim, heads, dim_heads, drop_out, window_size,
relative_pos_embedding, fuse_method='nai
v2xvit/models/sub_modules/mswin.py:84
Method__init__(self, dim, heads, num_types=2,
num_relations=4, dim_head=64, dropout=0.1)
v2xvit/models/sub_modules/hmsa.py:8
Method__init__(self, dim, depth, heads, dim_head, mlp_dim, dropout=0.)
v2xvit/models/sub_modules/base_transformer.py:84
Functioncorner2d_to_standup_box Find the minmaxx, minmaxy for each 2d box. (N, 4, 2) -> (N, 4) x1, y1, x2, y2 Parameters ---------- box2d : np.ndarray (
v2xvit/utils/box_utils.py:205