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(self