yolo2 keans++ nodejs package

环境

• npm >=6.4.1
• node >=8.12.0

介绍

K均值聚类算法： 该算法用来给多个散列点分类， 在yolo中用来归类多个label类型，使得训练收敛更加快。支持KMeans ++初始化

下载

npm install yolo2-kmeans-plus --save

使用

import KMeans from "k-mean-yolo2";

// 实例化Kmeans
const k = new KMeans({
    w: 416,
    h: 416
}, labels, {
    n_anchors: 5,
    loss_convergence: 0.000001,
    iterations_num: 100,
    plus: true,
    debug: false
});

// =>
// centroids: (5) [{…}, {…}, {…}, {…}, {…}]
// iterations: 4
// loss: 3.774831882178467
// map: (4) [Array(5), Array(5), Array(5), Array(5)]
// startCentroids: (5) [{…}, {…}, {…}, {…}, {…}]
console.log(k.compute_centroids());

实例参数说明

|方法|效果|属性| |----|-----|-----| |size|k-means尺寸|Object<{ w: number, h: number }>| |labels|标注数据|Array<{ w: number, h: number, x: number, y: number }>| |configs|配置参数|Object|

API方法说明

|方法|效果| |----|-----| |compute_centroids|获取K-Means计算结果|