yolo-helpers/dist/tensorflow/node.js

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

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.loadYoloModel = loadYoloModel;
const tf = __importStar(require("@tensorflow/tfjs-node"));
const promises_1 = require("fs/promises");
const common_1 = require("./common");
async function loadYoloModel(
/**
 * Can be with or without `/model.json`.
 *
 * Examples:
 * - "./saved_model/yolo11n-pose_web_model/model.json"
 * - "./saved_model/yolo11n-pose_web_model"
 * - "file://path/to/model.json"
 * - "http://localhost:8100/saved_models/yolo11n-pose_web_model"
 * - "https://domain.net/saved_models/yolo11n-pose_web_model/model.json"
 * */
modelPath) {
    if (!modelPath.endsWith('/model.json')) {
        modelPath = modelPath + '/model.json';
    }
    if (!modelPath.includes('://')) {
        modelPath = 'file://' + modelPath;
    }
    let model = tf.loadGraphModel(modelPath);
    let metadataPath = modelPath.replace(/model\.json$/, 'metadata.yaml');
    let metadata = loadText(metadataPath)
        .then(text => (0, common_1.parseMetadataYaml)(text))
        .catch((e) => {
        console.error(`failed to load ${metadataPath}, error:`, e);
        return {};
    });
    return (0, common_1.combineModelAndMetadata)(await model, await metadata);
}
async function loadText(path) {
    if (path.startsWith('file://')) {
        return await (0, promises_1.readFile)(path.slice('file://'.length), 'utf-8');
    }
    if (path.startsWith('http://') || path.startsWith('https://')) {
        return await (0, common_1.loadTextFromUrl)(path);
    }
    throw new Error(`Unsupported path: "${path}"`);
}