@tensorflow-models/body-pix
Version:
Pretrained BodyPix model in TensorFlow.js
5 lines (4 loc) • 515 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
export declare function getPointsConfidence(heatmapScores: tf.TensorBuffer<tf.Rank.R3>, heatMapCoords: tf.TensorBuffer<tf.Rank.R2>): Float32Array;
export declare function getOffsetVectors(heatMapCoordsBuffer: tf.TensorBuffer<tf.Rank.R2>, offsetsBuffer: tf.TensorBuffer<tf.Rank.R3>): tf.Tensor2D;
export declare function getOffsetPoints(heatMapCoordsBuffer: tf.TensorBuffer<tf.Rank.R2>, outputStride: number, offsetsBuffer: tf.TensorBuffer<tf.Rank.R3>): tf.Tensor2D;