Skip to content

Commit 2293f1f

Browse files
nikhilaravifacebook-github-bot
authored andcommitted
IoU for 3D boxes
Summary: I have implemented an exact solution for 3D IoU of oriented 3D boxes. This file includes: * box3d_overlap: which computes the exact IoU of box1 and box2 * box3d_overlap_sampling: which computes an approximate IoU of box1 and box2 by sampling points within the boxes Note that both implementations currently do not support batching. Our exact IoU implementation is based on the fact that the intersecting shape of the two 3D boxes will be formed by segments of the surface of the boxes. Our algorithm computes these segments by reasoning whether triangles of one box are within the second box and vice versa. We deal with intersecting triangles by clipping them. Reviewed By: gkioxari Differential Revision: D30667497 fbshipit-source-id: 2f747f410f90b7f854eeaf3036794bc3ac982917
1 parent 5b89c4e commit 2293f1f

File tree

3 files changed

+945
-3
lines changed

3 files changed

+945
-3
lines changed

pytorch3d/ops/sample_farthest_points.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
# LICENSE file in the root directory of this source tree.
66

77
from random import randint
8-
from typing import Optional, Tuple, Union, List
8+
from typing import List, Optional, Tuple, Union
99

1010
import torch
1111
from pytorch3d import _C

0 commit comments

Comments
 (0)