↓ 1 callersFunctionintersect We resize both tensors to [A,B,2] without new malloc: [A,2] -> [A,1,2] -> [A,B,2] [B,2] -> [1,B,2] -> [A,B,2] Then we compute the area of
models/functions/funcs.py:9
↓ 1 callersFunctionmask_iou Computes the pariwise mask IoU between two sets of masks of size [a, h, w] and [b, h, w]. The output is of size [a, b].
models/functions/funcs.py:58
↓ 1 callersFunctionmatrix_nms(cate_labels, seg_masks, sum_masks, cate_scores, sigma=2.0, kernel='gaussian')
models/functions/nms.py:15