rtmlib-ts 0.0.2

package/dist/solution/poseTracker.js

@@ -0,0 +1,106 @@
+/**
+ * PoseTracker - tracks poses across frames with cached detections
+ * Reduces detection frequency for better performance
+ */
+export class PoseTracker {
+    constructor(WholebodyClass, detFrequency = 7, toOpenpose = false, mode = 'balanced', backend = 'onnxruntime', device = 'cpu') {
+        this.cachedBoxes = [];
+        this.frameCount = 0;
+        this.nextId = 0;
+        this.detFrequency = detFrequency;
+        this.wholebody = new WholebodyClass(null, [640, 640], null, [288, 384], mode, toOpenpose, backend, device);
+    }
+    async init() {
+        await this.wholebody.init();
+    }
+    async call(image, imgWidth, imgHeight) {
+        this.frameCount++;
+        // Run detection periodically
+        if (this.frameCount % this.detFrequency === 0 || this.cachedBoxes.length === 0) {
+            const result = await this.wholebody.call(image, imgWidth, imgHeight);
+            this.updateCachedBoxes(result.keypoints, result.scores, imgWidth, imgHeight);
+        }
+        // Use cached boxes for pose estimation
+        const bboxes = this.cachedBoxes.map((tb) => tb.bbox);
+        const result = await this.wholebody.call(image, imgWidth, imgHeight, bboxes);
+        // Clean up old boxes
+        this.cleanupCachedBoxes();
+        return result;
+    }
+    updateCachedBoxes(keypoints, scores, imgWidth, imgHeight) {
+        // Simple tracking: create new boxes from keypoints
+        const newBoxes = [];
+        for (let i = 0; i < keypoints.length; i += 17) {
+            const instanceKeypoints = keypoints.slice(i, Math.min(i + 17, keypoints.length));
+            const instanceScores = scores.slice(i, Math.min(i + 17, scores.length));
+            // Calculate bounding box from keypoints
+            let minX = imgWidth;
+            let minY = imgHeight;
+            let maxX = 0;
+            let maxY = 0;
+            for (let j = 0; j < instanceKeypoints.length; j++) {
+                if (instanceScores[j] > 0.3) {
+                    const [x, y] = instanceKeypoints[j];
+                    minX = Math.min(minX, x);
+                    minY = Math.min(minY, y);
+                    maxX = Math.max(maxX, x);
+                    maxY = Math.max(maxY, y);
+                }
+            }
+            if (maxX > minX && maxY > minY) {
+                // Add padding
+                const padding = 0.25;
+                const width = maxX - minX;
+                const height = maxY - minY;
+                const paddedWidth = width * (1 + padding);
+                const paddedHeight = height * (1 + padding);
+                const centerX = (minX + maxX) / 2;
+                const centerY = (minY + maxY) / 2;
+                const x1 = Math.max(0, centerX - paddedWidth / 2);
+                const y1 = Math.max(0, centerY - paddedHeight / 2);
+                const x2 = Math.min(imgWidth, centerX + paddedWidth / 2);
+                const y2 = Math.min(imgHeight, centerY + paddedHeight / 2);
+                // Try to match with existing boxes
+                let matched = false;
+                for (const cachedBox of this.cachedBoxes) {
+                    const iou = this.calculateIoU([x1, y1, x2, y2], [cachedBox.bbox.x1, cachedBox.bbox.y1, cachedBox.bbox.x2, cachedBox.bbox.y2]);
+                    if (iou > 0.3) {
+                        cachedBox.bbox = { x1, y1, x2, y2 };
+                        cachedBox.age = 0;
+                        cachedBox.lastSeen = this.frameCount;
+                        newBoxes.push(cachedBox);
+                        matched = true;
+                        break;
+                    }
+                }
+                if (!matched) {
+                    newBoxes.push({
+                        bbox: { x1, y1, x2, y2 },
+                        age: 0,
+                        lastSeen: this.frameCount,
+                        id: this.nextId++,
+                    });
+                }
+            }
+        }
+        this.cachedBoxes = newBoxes;
+    }
+    calculateIoU(box1, box2) {
+        const x1 = Math.max(box1[0], box2[0]);
+        const y1 = Math.max(box1[1], box2[1]);
+        const x2 = Math.min(box1[2], box2[2]);
+        const y2 = Math.min(box1[3], box2[3]);
+        const interWidth = Math.max(0, x2 - x1);
+        const interHeight = Math.max(0, y2 - y1);
+        const interArea = interWidth * interHeight;
+        const box1Area = (box1[2] - box1[0]) * (box1[3] - box1[1]);
+        const box2Area = (box2[2] - box2[0]) * (box2[3] - box2[1]);
+        const unionArea = box1Area + box2Area - interArea;
+        return unionArea > 0 ? interArea / unionArea : 0;
+    }
+    cleanupCachedBoxes() {
+        // Remove boxes that haven't been seen for too long
+        const maxAge = this.detFrequency * 3;
+        this.cachedBoxes = this.cachedBoxes.filter((box) => this.frameCount - box.lastSeen < maxAge);
+    }
+}

package/dist/solution/wholebody.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Wholebody solution - combines detection and pose estimation
+ * Based on rtmlib Wholebody class
+ * Supports YOLO12 and YOLOX detectors with RTMW pose model
+ */
+import { BBox, ModeType } from '../types';
+export declare class Wholebody {
+    private detModel;
+    private poseModel;
+    private detectorType;
+    private static readonly MODE;
+    constructor(det?: string | null, detInputSize?: [number, number], pose?: string | null, poseInputSize?: [number, number], mode?: ModeType, toOpenpose?: boolean, backend?: 'onnxruntime', device?: string, detectorType?: 'yolox' | 'yolo12');
+    init(): Promise<void>;
+    call(image: Uint8Array, imgWidth: number, imgHeight: number, bboxes?: BBox[]): Promise<{
+        keypoints: number[][];
+        scores: number[];
+    }>;
+}
+//# sourceMappingURL=wholebody.d.ts.map

package/dist/solution/wholebody.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wholebody.d.ts","sourceRoot":"","sources":["../../src/solution/wholebody.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAe,MAAM,UAAU,CAAC;AAEvD,qBAAa,SAAS;IACpB,OAAO,CAAC,QAAQ,CAAiB;IACjC,OAAO,CAAC,SAAS,CAAU;IAC3B,OAAO,CAAC,YAAY,CAAqB;IAEzC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAmB1B;gBAGA,GAAG,GAAE,MAAM,GAAG,IAAW,EACzB,YAAY,GAAE,CAAC,MAAM,EAAE,MAAM,CAAc,EAC3C,IAAI,GAAE,MAAM,GAAG,IAAW,EAC1B,aAAa,GAAE,CAAC,MAAM,EAAE,MAAM,CAAc,EAC5C,IAAI,GAAE,QAAqB,EAC3B,UAAU,GAAE,OAAe,EAC3B,OAAO,GAAE,aAA6B,EACtC,MAAM,GAAE,MAAc,EACtB,YAAY,GAAE,OAAO,GAAG,QAAkB;IA+CtC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAKrB,IAAI,CACR,KAAK,EAAE,UAAU,EACjB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,IAAI,EAAE,GACd,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;CA0BxD"}

package/dist/solution/wholebody.js

@@ -0,0 +1,82 @@
+/**
+ * Wholebody solution - combines detection and pose estimation
+ * Based on rtmlib Wholebody class
+ * Supports YOLO12 and YOLOX detectors with RTMW pose model
+ */
+import { YOLOX } from '../models/yolox';
+import { YOLO12 } from '../models/yolo12';
+import { RTMPose } from '../models/rtmpose';
+export class Wholebody {
+    constructor(det = null, detInputSize = [640, 640], pose = null, poseInputSize = [384, 288], mode = 'balanced', toOpenpose = false, backend = 'onnxruntime', device = 'cpu', detectorType = 'yolox') {
+        this.detectorType = detectorType;
+        // Use mode config if det/pose not specified
+        let finalDet = det;
+        let finalDetInputSize = detInputSize;
+        let finalPose = pose;
+        let finalPoseInputSize = poseInputSize;
+        if (det === null) {
+            finalDet = Wholebody.MODE[mode].det;
+            finalDetInputSize = Wholebody.MODE[mode].detInputSize;
+        }
+        if (pose === null) {
+            finalPose = Wholebody.MODE[mode].pose;
+            finalPoseInputSize = Wholebody.MODE[mode].poseInputSize;
+        }
+        // Initialize detector based on type
+        if (detectorType === 'yolo12') {
+            this.detModel = new YOLO12(finalDet, finalDetInputSize, 0.45, 0.5, backend);
+        }
+        else {
+            this.detModel = new YOLOX(finalDet, finalDetInputSize, 0.45, 0.7, backend);
+        }
+        this.poseModel = new RTMPose(finalPose, finalPoseInputSize, toOpenpose, backend);
+    }
+    async init() {
+        await this.detModel.init();
+        await this.poseModel.init();
+    }
+    async call(image, imgWidth, imgHeight, bboxes) {
+        // Run detection if bboxes not provided
+        let finalBboxes = bboxes;
+        if (!finalBboxes || finalBboxes.length === 0) {
+            const detections = await this.detModel.call(image, imgWidth, imgHeight);
+            // Convert Detection[] or BBox[] to BBox[]
+            finalBboxes = detections.map((d) => {
+                if ('bbox' in d) {
+                    // Detection type
+                    return {
+                        x1: d.bbox.x1,
+                        y1: d.bbox.y1,
+                        x2: d.bbox.x2,
+                        y2: d.bbox.y2,
+                    };
+                }
+                // Already BBox type
+                return d;
+            });
+        }
+        // Run pose estimation
+        const result = await this.poseModel.call(image, imgWidth, imgHeight, finalBboxes);
+        return result;
+    }
+}
+Wholebody.MODE = {
+    performance: {
+        det: 'https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/yolox_m_8xb8-300e_humanart-c2c7a14a.zip',
+        detInputSize: [640, 640],
+        pose: 'https://download.openmmlab.com/mmpose/v1/projects/rtmw/onnx_sdk/rtmw-dw-x-l_simcc-cocktail14_270e-384x288_20231122.zip',
+        poseInputSize: [384, 288],
+    },
+    lightweight: {
+        det: 'https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/yolox_tiny_8xb8-300e_humanart-6f3252f9.zip',
+        detInputSize: [416, 416],
+        pose: 'https://download.openmmlab.com/mmpose/v1/projects/rtmw/onnx_sdk/rtmw-dw-l-m_simcc-cocktail14_270e-256x192_20231122.zip',
+        poseInputSize: [256, 192],
+    },
+    balanced: {
+        det: 'https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/yolox_m_8xb8-300e_humanart-c2c7a14a.zip',
+        detInputSize: [640, 640],
+        pose: 'https://download.openmmlab.com/mmpose/v1/projects/rtmw/onnx_sdk/rtmw-dw-x-l_simcc-cocktail14_270e-256x192_20231122.zip',
+        poseInputSize: [256, 192],
+    },
+};

package/dist/solution/wholebody3d.d.ts

@@ -0,0 +1,22 @@
+/**
+ * Wholebody3D solution - combines detection and 3D pose estimation
+ * Based on rtmlib Wholebody3d class
+ * Uses YOLOX detector with RTMW3D pose model
+ */
+import { BBox, ModeType, BackendType } from '../types';
+export interface Wholebody3DResult {
+    keypoints: number[][][];
+    scores: number[][];
+    keypointsSimcc: number[][][];
+    keypoints2d: number[][][];
+}
+export declare class Wholebody3D {
+    private detModel;
+    private poseModel;
+    private static readonly MODE;
+    constructor(det?: string | null, detInputSize?: [number, number], pose?: string | null, poseInputSize?: [number, number], // [width=288, height=384]
+    mode?: ModeType, toOpenpose?: boolean, backend?: BackendType);
+    init(): Promise<void>;
+    call(image: Uint8Array, imgWidth: number, imgHeight: number, bboxes?: BBox[]): Promise<Wholebody3DResult>;
+}
+//# sourceMappingURL=wholebody3d.d.ts.map

package/dist/solution/wholebody3d.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wholebody3d.d.ts","sourceRoot":"","sources":["../../src/solution/wholebody3d.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAe,WAAW,EAAE,MAAM,UAAU,CAAC;AAEpE,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACxB,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC;IACnB,cAAc,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IAC7B,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;CAC3B;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAQ;IACxB,OAAO,CAAC,SAAS,CAAY;IAE7B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAmB1B;gBAGA,GAAG,GAAE,MAAM,GAAG,IAAW,EACzB,YAAY,GAAE,CAAC,MAAM,EAAE,MAAM,CAAc,EAC3C,IAAI,GAAE,MAAM,GAAG,IAAW,EAC1B,aAAa,GAAE,CAAC,MAAM,EAAE,MAAM,CAAc,EAAG,0BAA0B;IACzE,IAAI,GAAE,QAAqB,EAC3B,UAAU,GAAE,OAAe,EAC3B,OAAO,GAAE,WAAsB;IAkC3B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAKrB,IAAI,CACR,KAAK,EAAE,UAAU,EACjB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,IAAI,EAAE,GACd,OAAO,CAAC,iBAAiB,CAAC;CA+B9B"}

package/dist/solution/wholebody3d.js

@@ -0,0 +1,75 @@
+/**
+ * Wholebody3D solution - combines detection and 3D pose estimation
+ * Based on rtmlib Wholebody3d class
+ * Uses YOLOX detector with RTMW3D pose model
+ */
+import { YOLOX } from '../models/yolox';
+import { RTMPose3D } from '../models/rtmpose3d';
+export class Wholebody3D {
+    constructor(det = null, detInputSize = [640, 640], pose = null, poseInputSize = [288, 384], // [width=288, height=384]
+    mode = 'balanced', toOpenpose = false, backend = 'webgpu') {
+        // Use mode config if det/pose not specified
+        let finalDet = det;
+        let finalDetInputSize = detInputSize;
+        let finalPose = pose;
+        let finalPoseInputSize = poseInputSize;
+        if (det === null) {
+            finalDet = Wholebody3D.MODE[mode].det;
+            finalDetInputSize = Wholebody3D.MODE[mode].detInputSize;
+        }
+        if (pose === null) {
+            finalPose = Wholebody3D.MODE[mode].pose;
+            finalPoseInputSize = Wholebody3D.MODE[mode].poseInputSize;
+        }
+        this.detModel = new YOLOX(finalDet, finalDetInputSize, 0.45, 0.7, backend);
+        this.poseModel = new RTMPose3D(finalPose, finalPoseInputSize, toOpenpose, backend);
+    }
+    async init() {
+        await this.detModel.init();
+        await this.poseModel.init();
+    }
+    async call(image, imgWidth, imgHeight, bboxes) {
+        // Run detection if bboxes not provided
+        let finalBboxes = bboxes;
+        if (!finalBboxes || finalBboxes.length === 0) {
+            const detections = await this.detModel.call(image, imgWidth, imgHeight);
+            // Convert Detection[] or BBox[] to BBox[]
+            finalBboxes = detections.map((d) => {
+                if ('bbox' in d) {
+                    // Detection type
+                    return {
+                        x1: d.bbox.x1,
+                        y1: d.bbox.y1,
+                        x2: d.bbox.x2,
+                        y2: d.bbox.y2,
+                    };
+                }
+                // Already BBox type
+                return d;
+            });
+        }
+        // Run 3D pose estimation
+        const result = await this.poseModel.call(image, imgWidth, imgHeight, finalBboxes);
+        return result;
+    }
+}
+Wholebody3D.MODE = {
+    performance: {
+        det: 'https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/yolox_m_8xb8-300e_humanart-c2c7a14a.zip',
+        detInputSize: [640, 640],
+        pose: 'https://huggingface.co/Soykaf/RTMW3D-x/resolve/main/onnx/rtmw3d-x_8xb64_cocktail14-384x288-b0a0eab7_20240626.onnx',
+        poseInputSize: [288, 384], // [width=288, height=384] - creates tensor [1,3,384,288]
+    },
+    lightweight: {
+        det: 'https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/yolox_tiny_8xb8-300e_humanart-6f3252f9.zip',
+        detInputSize: [416, 416],
+        pose: 'https://huggingface.co/Soykaf/RTMW3D-x/resolve/main/onnx/rtmw3d-x_8xb64_cocktail14-384x288-b0a0eab7_20240626.onnx',
+        poseInputSize: [192, 256], // [width=192, height=256]
+    },
+    balanced: {
+        det: 'https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/yolox_m_8xb8-300e_humanart-c2c7a14a.zip',
+        detInputSize: [640, 640],
+        pose: 'https://huggingface.co/Soykaf/RTMW3D-x/resolve/main/onnx/rtmw3d-x_8xb64_cocktail14-384x288-b0a0eab7_20240626.onnx',
+        poseInputSize: [288, 384], // [width=288, height=384] - creates tensor [1,3,384,288]
+    },
+};

package/dist/types/index.d.ts

@@ -0,0 +1,52 @@
+/**
+ * Basic types for rtmlib-ts
+ * Based on rtmlib Python library
+ */
+export interface Keypoint {
+    x: number;
+    y: number;
+    score: number;
+    id: number;
+}
+export interface BodyResult {
+    keypoints: Array<Keypoint | null>;
+    totalScore: number;
+    totalParts: number;
+}
+export type HandResult = Keypoint[];
+export type FaceResult = Keypoint[];
+export interface PoseResult {
+    body: BodyResult;
+    leftHand: HandResult | null;
+    rightHand: HandResult | null;
+    face: FaceResult | null;
+}
+export interface BBox {
+    x1: number;
+    y1: number;
+    x2: number;
+    y2: number;
+}
+export interface Detection {
+    bbox: BBox;
+    score: number;
+    classId: number;
+}
+export interface ModelConfig {
+    det: string;
+    detInputSize: [number, number];
+    pose: string;
+    poseInputSize: [number, number];
+}
+export type ModeType = 'performance' | 'lightweight' | 'balanced';
+export type BackendType = 'opencv' | 'onnxruntime' | 'openvino' | 'wasm' | 'webgpu';
+export type DeviceType = 'cpu' | 'cuda' | 'mps' | string;
+export interface ImageData {
+    data: Uint8Array;
+    width: number;
+    height: number;
+    channels: number;
+}
+export type RGBImage = ImageData;
+export type BGRImage = ImageData;
+//# sourceMappingURL=index.d.ts.map

package/dist/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,MAAM,WAAW,QAAQ;IACvB,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;IACV,KAAK,EAAE,MAAM,CAAC;IACd,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,CAAC;IAClC,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,UAAU,GAAG,QAAQ,EAAE,CAAC;AACpC,MAAM,MAAM,UAAU,GAAG,QAAQ,EAAE,CAAC;AAEpC,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE,UAAU,GAAG,IAAI,CAAC;IAC5B,SAAS,EAAE,UAAU,GAAG,IAAI,CAAC;IAC7B,IAAI,EAAE,UAAU,GAAG,IAAI,CAAC;CACzB;AAED,MAAM,WAAW,IAAI;IACnB,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACjC;AAED,MAAM,MAAM,QAAQ,GAAG,aAAa,GAAG,aAAa,GAAG,UAAU,CAAC;AAElE,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,aAAa,GAAG,UAAU,GAAG,MAAM,GAAG,QAAQ,CAAC;AACpF,MAAM,MAAM,UAAU,GAAG,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;AAEzD,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,MAAM,QAAQ,GAAG,SAAS,CAAC;AACjC,MAAM,MAAM,QAAQ,GAAG,SAAS,CAAC"}

package/dist/types/index.js

@@ -0,0 +1,5 @@
+/**
+ * Basic types for rtmlib-ts
+ * Based on rtmlib Python library
+ */
+export {};

package/dist/visualization/draw.d.ts

@@ -0,0 +1,57 @@
+/**
+ * Drawing utilities for visualization
+ * Based on rtmlib Python library
+ */
+/**
+ * Draw bounding boxes on image
+ */
+export declare function drawBbox(img: Uint8Array, width: number, height: number, bboxes: Array<[number, number, number, number]>, color?: [number, number, number]): Uint8Array;
+/**
+ * Draw skeleton on image
+ */
+export declare function drawSkeleton(img: Uint8Array, width: number, height: number, keypoints: number[][], scores: number[], openposeSkeleton?: boolean, kptThr?: number, radius?: number, lineWidth?: number): Uint8Array;
+/**
+ * Draw detections on HTML Canvas
+ * @param ctx - Canvas 2D context
+ * @param detections - Array of detected objects
+ * @param color - Base color for boxes (default: green)
+ */
+export declare function drawDetectionsOnCanvas(ctx: CanvasRenderingContext2D, detections: Array<{
+    bbox: {
+        x1: number;
+        y1: number;
+        x2: number;
+        y2: number;
+        confidence: number;
+    };
+    className?: string;
+}>, color?: string): void;
+/**
+ * Draw pose skeleton on HTML Canvas
+ * @param ctx - Canvas 2D context
+ * @param people - Array of people with keypoints
+ * @param confidenceThreshold - Minimum keypoint confidence to draw (default: 0.3)
+ */
+export declare function drawPoseOnCanvas(ctx: CanvasRenderingContext2D, people: Array<{
+    bbox: {
+        x1: number;
+        y1: number;
+        x2: number;
+        y2: number;
+        confidence: number;
+    };
+    keypoints: Array<{
+        x: number;
+        y: number;
+        score: number;
+        visible: boolean;
+    }>;
+}>, confidenceThreshold?: number): void;
+/**
+ * Draw both detections and pose on canvas (convenience method)
+ * @param ctx - Canvas 2D context
+ * @param results - Detection or pose results
+ * @param mode - 'object' or 'pose'
+ */
+export declare function drawResultsOnCanvas(ctx: CanvasRenderingContext2D, results: any[], mode?: 'object' | 'pose'): void;
+//# sourceMappingURL=draw.d.ts.map

package/dist/visualization/draw.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"draw.d.ts","sourceRoot":"","sources":["../../src/visualization/draw.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAkBH;;GAEG;AACH,wBAAgB,QAAQ,CACtB,GAAG,EAAE,UAAU,EACf,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,KAAK,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,EAC/C,KAAK,GAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAe,GAC5C,UAAU,CAgDZ;AAED;;GAEG;AACH,wBAAgB,YAAY,CAC1B,GAAG,EAAE,UAAU,EACf,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EAAE,EAAE,EACrB,MAAM,EAAE,MAAM,EAAE,EAChB,gBAAgB,GAAE,OAAe,EACjC,MAAM,GAAE,MAAY,EACpB,MAAM,GAAE,MAAU,EAClB,SAAS,GAAE,MAAU,GACpB,UAAU,CAoDZ;AAqPD;;;;;GAKG;AACH,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,wBAAwB,EAC7B,UAAU,EAAE,KAAK,CAAC;IAChB,IAAI,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7E,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC,EACF,KAAK,GAAE,MAAkB,GACxB,IAAI,CAsBN;AAED;;;;;GAKG;AACH,wBAAgB,gBAAgB,CAC9B,GAAG,EAAE,wBAAwB,EAC7B,MAAM,EAAE,KAAK,CAAC;IACZ,IAAI,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7E,SAAS,EAAE,KAAK,CAAC;QAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;CAC7E,CAAC,EACF,mBAAmB,GAAE,MAAY,GAChC,IAAI,CAoFN;AAED;;;;;GAKG;AACH,wBAAgB,mBAAmB,CACjC,GAAG,EAAE,wBAAwB,EAC7B,OAAO,EAAE,GAAG,EAAE,EACd,IAAI,GAAE,QAAQ,GAAG,MAAiB,GACjC,IAAI,CAMN"}