face-api/build/dom/toNetInput.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.toNetInput = void 0;
const utils_1 = require("../utils");
const awaitMediaLoaded_1 = require("./awaitMediaLoaded");
const isMediaElement_1 = require("./isMediaElement");
const NetInput_1 = require("./NetInput");
const resolveInput_1 = require("./resolveInput");
/**
 * Validates the input to make sure, they are valid net inputs and awaits all media elements
 * to be finished loading.
 *
 * @param input The input, which can be a media element or an array of different media elements.
 * @returns A NetInput instance, which can be passed into one of the neural networks.
 */
async function toNetInput(inputs) {
    if (inputs instanceof NetInput_1.NetInput) {
        return inputs;
    }
    let inputArgArray = Array.isArray(inputs)
        ? inputs
        : [inputs];
    if (!inputArgArray.length) {
        throw new Error('toNetInput - empty array passed as input');
    }
    const getIdxHint = (idx) => Array.isArray(inputs) ? ` at input index ${idx}:` : '';
    const inputArray = inputArgArray.map(resolveInput_1.resolveInput);
    inputArray.forEach((input, i) => {
        if (!isMediaElement_1.isMediaElement(input) && !utils_1.isTensor3D(input) && !utils_1.isTensor4D(input)) {
            if (typeof inputArgArray[i] === 'string') {
                throw new Error(`toNetInput -${getIdxHint(i)} string passed, but could not resolve HTMLElement for element id ${inputArgArray[i]}`);
            }
            throw new Error(`toNetInput -${getIdxHint(i)} expected media to be of type HTMLImageElement | HTMLVideoElement | HTMLCanvasElement | tf.Tensor3D, or to be an element id`);
        }
        if (utils_1.isTensor4D(input)) {
            // if tf.Tensor4D is passed in the input array, the batch size has to be 1
            const batchSize = input.shape[0];
            if (batchSize !== 1) {
                throw new Error(`toNetInput -${getIdxHint(i)} tf.Tensor4D with batchSize ${batchSize} passed, but not supported in input array`);
            }
        }
    });
    // wait for all media elements being loaded
    await Promise.all(inputArray.map(input => isMediaElement_1.isMediaElement(input) && awaitMediaLoaded_1.awaitMediaLoaded(input)));
    return new NetInput_1.NetInput(inputArray, Array.isArray(inputs));
}
exports.toNetInput = toNetInput;
//# sourceMappingURL=toNetInput.js.map