yolo-helpers/dist/yolo-pose/node.d.ts

Helper functions to use models converted from YOLO in browser and Node.js

import * as tf from '@tensorflow/tfjs-node';
import { DecodePoseArgs } from './common';
import { ImageInput } from '../tensorflow/node';
export * from './common';
export type DetectPoseArgs = {
    model: tf.InferenceModel;
    /** used for image resize when necessary, auto inferred from model shape */
    input_shape?: {
        width: number;
        height: number;
    };
} & Omit<DecodePoseArgs, 'output'> & ImageInput;
/**
 * box features:
 *   - x, y, width, height
 *   - highest confidence, class_index
 *   - keypoints
 *
 * keypoint features:
 *   - x, y, visibility
 *
 * The x, y, width, height are in pixel unit, NOT normalized in the range of [0, 1].
 * The the pixel units are scaled to the input_shape.
 *
 * The confidence are already normalized between 0 to 1.
 */
export declare function detectPose(args: DetectPoseArgs): Promise<import("./common").PoseResult>;
/**
 * Sync version of `detectPose`.
 */
export declare function detectPoseSync(args: DetectPoseArgs): import("./common").PoseResult;