anchor设置 yolo_[DL]YOLO中anchor box是如何通过聚类确定的

from os import listdir

from os.path import isfile, join

import argparse

#import cv2

import numpy as np

import sys

import os

import shutil

import random

import math

def IOU(x,centroids):

'''

:param x: 某一个ground truth的w,h

:param centroids: anchor的w,h的集合[(w,h),(),...]，共k个

:return: 单个ground truth box与所有k个anchor box的IoU值集合

'''

IoUs = []

w, h = x # ground truth的w,h

for centroid in centroids:

c_w,c_h = centroid #anchor的w,h

if c_w>=w and c_h>=h: #anchor包围ground truth

iou = w*h/(c_w*c_h)

elif c_w>=w and c_h<=h: #anchor宽矮

iou = w*c_h/(w*h + (c_w-w)*c_h)

elif c_w<=w and c_h>=h: #anchor瘦长

iou = c_w*h/(w*h + c_w*(c_h-h))

else: #ground truth包围anchor means both w,h are bigger than c_w and c_h respectively

iou = (c_w*c_h)/(w*h)

IoUs.append(iou) # will become (k,) shape

return np.array(IoUs)

def avg_IOU(X,centroids):

'''

:param X: ground truth的w,h的集合[(w,h),(),...]

:param centroids: anchor的w,h的集合[(w,h),(),...]，共k个

'''

n,d = X.shape

sum = 0.

for i in range(X.shape[0]):

sum+= max(IOU(X[i],centroids)) #返回一