24 lines
1.2 KiB
JavaScript
24 lines
1.2 KiB
JavaScript
|
/* 导出运动学 + 胶囊碰撞模型给 PyTorch（ml/common/kin_torch.py 读取）
|
|||
|
|
* node ml/prep/export_kin.cjs [out=ml/assets/kin.json]
|
|||
|
|
* collide.json 重新拟合/重生成 pair2d 后必须重跑（指纹不符时 Python 端拒绝加载）。 */
|
|||
|
|
'use strict';
|
|||
|
|
const fs = require('fs'), path = require('path');
|
|||
|
|
const { S } = require('../../tests/route_bench_setup.cjs')();
|
|||
|
|
const OUT = process.argv[2] || path.join(__dirname, '../assets/kin.json');
|
|||
|
|
const COL = JSON.parse(fs.readFileSync(path.join(__dirname, '../../docs/collide.json'), 'utf8'));
|
|||
|
|
const ce = S.clusterExport();
|
|||
|
|
const out = {
|
|||
|
|
fp: S.modelFingerprint(),
|
|||
|
|
order: S.ORDER,
|
|||
|
|
P: S.ORDER.map(id => S.P[id]), D: S.ORDER.map(id => S.D[id]),
|
|||
|
|
lim: S.ORDER.map(id => [S.LIM[id].lo, S.LIM[id].hi, S.LIM[id].cont ? 1 : 0]),
|
|||
|
|
tcp0: S.TCP0, capm: S.CAPM,
|
|||
|
|
caps: COL.caps.map(c => ({ g: c.g, a: c.a, b: c.b, r: c.r })),
|
|||
|
|
clusters: ce.clusters, cpairs: ce.cpairs,
|
|||
|
|
lever: S.leverBounds(20000),
|
|||
|
|
generated: new Date().toISOString()
|
|||
|
|
};
|
|||
|
|
fs.mkdirSync(path.dirname(OUT), { recursive: true });
|
|||
|
|
fs.writeFileSync(OUT, JSON.stringify(out));
|
|||
|
|
console.log('wrote', OUT, 'caps', out.caps.length, 'clusters', out.clusters.length, 'cpairs', out.cpairs.length, 'fp', out.fp);
|