redisai-js/lib/helpers.js

A high-performance Typescript client for RedisAI

"use strict";
Object.defineProperty(exports, "__esModule", {value: true});
var Helpers = /** @class */ (function () {
    function Helpers() {
    }

    Helpers.prototype.argmax = function (arr) {
        var index = 0;
        var value = arr[0];
        arr.forEach(function (item, key) {
            if (item > value) {
                value = item;
                index = key;
            }
        });
        return index;
    };
    Helpers.prototype.normalizeRGB = function (buffer) {
        var npixels = buffer.length / 4;
        var out = new Float32Array(npixels * 3);
        for (var i = 0; i < npixels; i++) {
            out[3 * i] = buffer[4 * i] / 128 - 1;
            out[3 * i + 1] = buffer[4 * i + 1] / 128 - 1;
            out[3 * i + 2] = buffer[4 * i + 2] / 128 - 1;
        }
        return out;
    };
    Helpers.prototype.bufferToFloat32Array = function (buffer) {
        var out_array = new Float32Array(buffer.length / 4);
        for (var i = 0; i < out_array.length; i++) {
            out_array[i] = buffer.readFloatLE(4 * i);
        }
        return out_array;
    };
    return Helpers;
}());
exports.Helpers = Helpers;
//# sourceMappingURL=Helpers.js.map