yolo voc锚框聚类

0，yolo 锚框聚类思想

随机选择锚框个数，原论文数量为5
以此遍历每个框，进行聚类

方法缺点，容易收到初始值影响，聚类方法不稳定

1， 代码

# *_*coding:utf-8 *_*
import sys
from xml.etree import ElementTree
from xml.etree.ElementTree import Element, SubElement
from lxml import etree
import numpy as np
import os
import sys
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans

class PascalVocReader:
    def __init__(self, path_xml, height, width, num=5):
        self.path_xml = path_xml
        self.height = height
        self.width = width
        self.km = KMeans(n_clusters=num, init="k-means++", n_init=10, max_iter=3000000, tol=1e-3, random_state=0)

        try:
            self.get_points()
        except:
            pass

    def get_points(self):
        self.data = []
        for file in os.listdir