@tensorflow-models/body-pix
Version:
Pretrained BodyPix model in TensorFlow.js
5 lines (4 loc) • 368 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
export declare function toMaskTensor(segmentScores: tf.Tensor2D, threshold: number): tf.Tensor2D;
export declare function decodePartSegmentation(segmentationMask: tf.Tensor2D, partHeatmapScores: tf.Tensor3D): tf.Tensor2D;
export declare function decodeOnlyPartSegmentation(partHeatmapScores: tf.Tensor3D): tf.Tensor2D;