1.0.0 • Published 4 years ago
k-means-yolo2 v1.0.0
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计算结果 |
1.0.0
4 years ago