Yolov2 聚类生成先验框代码详解（Yolov3同理）----自用笔记

原创

于 2021-11-13 21:04:57 发布

· 1.6k 阅读

12 ·

版权

文章标签：

#聚类

代码来自https://blog.youkuaiyun.com/weixin_44791964/article/details/102687531

完整代码如下：

import numpy as np
import xml.etree.ElementTree as ET
import glob
import random

def cas_iou(box,cluster):
    x = np.minimum(cluster[:,0],box[0])
    y = np.minimum(cluster[:,1],box[1])

    intersection = x * y
    area1 = box[0] * box[1]

    area2 = cluster[:,0] * cluster[:,1]
    iou = intersection / (area1 + area2 -intersection)

    return iou

def avg_iou(box,cluster):
    return np.mean([np.max(cas_iou(box[i],cluster)) for i in range(box.shape[

最低0.47元/天解锁文章