rwightman

import torch
from torch import nn
import torch.nn.functional as F
from collections import OrderedDict
import math

def pdist(v):
    dist = torch.norm(v[:, None] - v, dim=2, p=2)
    return dist

meyerjo

def bb_intersection_over_union(boxA, boxB):
    # determine the (x, y)-coordinates of the intersection rectangle
    xA = max(boxA[0], boxB[0])
    yA = max(boxA[1], boxB[1])
    xB = min(boxA[2], boxB[2])
    yB = min(boxA[3], boxB[3])

    # compute the area of intersection rectangle
    interArea = abs(max((xB - xA, 0)) * max((yB - yA), 0))
    if interArea == 0:

aparrish