从posenet提取关键点到json文件?

时间:2019-04-05 10:45:14

标签: javascript json tensorflow tensorflow.js pose-estimation

我正在研究posenet的tensorflow实现,以实时进行姿势估计,如果可能的话,也可以在离线模式下进行姿势估计。我正在研究以下回购协议:

  

https://github.com/tensorflow/tfjs-models/tree/master/posenet

在下面的代码部分的以下函数中读取关键点

    export function drawKeypoints(keypoints, minConfidence, ctx, scale = 1) {
  for (let i = 0; i < keypoints.length; i++) {
    const keypoint = keypoints[i];

    if (keypoint.score < minConfidence) {
      continue;
    }

    const {y, x} = keypoint.position;
    drawPoint(ctx, y * scale, x * scale, 3, color);
  }
}

https://github.com/tensorflow/tfjs-models/blob/master/posenet/demos/demo_util.js

我正在研究将关键点提取到json文件的可能性,如果可能的话?

任何有关此方面的提示都将非常有帮助。

1 个答案:

答案 0 :(得分:0)

将大块保存到文件是在浏览器环境中写入文件的常用方法。在nodejs服务器中,可以为此目的使用fs模块。

这是在浏览器中完成此操作的一种方法

   var blob = new Blob( [ keypointsData ], {
    type: 'application/octet-stream'
});

var url = URL.createObjectURL( blob );
var link = document.createElement( 'a' );
link.setAttribute( 'href', url );
link.setAttribute( 'download', 'data.json' );
var event = document.createEvent( 'MouseEvents' );
event.initMouseEvent( 'click', true, true, window, 1, 0, 0, 0, 0, false, false, false, false, 0, null);
link.dispatchEvent( event );