keras-yolov3项目之四：kmeans.py文件

kmeans.py文件主要是对训练数据集的标注框进行聚类，最终输出9个标注框，并将聚类好的9个标注框存储在yolo_anchors.txt文件中。

import numpy as np class YOLO_Kmeans: """ YOLO_Kmeans聚类模型 """ def __init__(self, cluster_number, filename): """ 初始化参数 参数介绍： cluster_number：类别数量 filename：文件名 "