pytorch 批量 iou

目录

pytorch批量max

pytorch批量iou

python 单个iou，批量iou

---

pytorch批量max

a = torch.Tensor([[random.randint(0, 20), random.randint(0, 20), random.randint(0, 20)]])
b1_x1=torch.Tensor([[random.randint(0, 20),random.randint(0, 20), random.randint(0, 20)]])
inter_rect_x1 = torch.max(b1_x1, a)
print(a)
print(b1_x1)
print(inter_rect_x1)

pytorch批量iou

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import datetime

import torch
from numba import jit
import numpy as np

import random


def bbox_iou(box1, box2, x1y1x2y2=True):
    """
    Returns the IoU of two bounding boxes
    """
    if