Bounding Box 의 IOU 계산 코드 두 개

1365 단어
Bounding Box 의 데이터 구 조 는 (xmin, ymin, xmax, ymax) 이다.
입력: box 1, box 2
출력: IOU 값
import numpy as np
def iou(box1,box2):
    assert box1.size()==4 and box2.size()==4,"bounding box coordinate size must be 4"
      bxmin = np.max(box1[0],box2[0])
      bymin = np.max(box1[1],box2[1])
      bxmax = np.min(box1[2],box2[2])
      bymax = np.min(box1[3],box2[3])
      bwidth = bxmax-bxmin
      bhight = bymax-bxmin
      inter = bwidth*bhight
      union = (box1[2]-box1[0])*(box1[3]-box1[1])+(box2[2]-box2[0])*(box2[3]-box2[1])-inter
      return inter/union

 
     
 
다음으로 전송:https://www.cnblogs.com/houjun/p/10454166.html

좋은 웹페이지 즐겨찾기