redisai-js
Version:
A high-performance Typescript client for RedisAI
50 lines • 1.46 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.Helpers = void 0;
var Helpers = /** @class */ (function () {
function Helpers() {
}
/**
* Returns the index of the first occurrence of the maximum value along an array.
* @param arr
*/
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;
};
/**
* TODO: document
* @param buffer
*/
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;
};
/**
* TODO: document
* @param buffer
*/
Helpers.prototype.bufferToFloat32Array = function (buffer) {
var outArray = new Float32Array(buffer.length / 4);
for (var i = 0; i < outArray.length; i++) {
outArray[i] = buffer.readFloatLE(4 * i);
}
return outArray;
};
return Helpers;
}());
exports.Helpers = Helpers;
//# sourceMappingURL=helpers.js.map