tf-gaussian-blur/build/index.js

A small gaussian blur library for tensorflow.js (tf.js)

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.blur = exports.getGaussianKernel = void 0;
var tf = require("@tensorflow/tfjs");
function get1dGaussianKernel(sigma, size) {
    // Generate a 1d gaussian distribution size numbers long
    var range = tf.range(Math.floor(-size / 2) + 1, Math.floor(size / 2) + 1);
    var distribution = tf.pow(tf.exp(range.div(-2.0 * (sigma * sigma))), 2);
    var normalized = distribution.div(tf.sum(distribution));
    return normalized;
}
function get2dGaussianKernel(size, sigma) {
    // This default is to mimic opencv2.
    sigma = sigma === undefined ? 0.3 * ((size - 1) * 0.5 - 1) + 0.8 : sigma;
    var kerne1d = get1dGaussianKernel(sigma, size);
    return tf.outerProduct(kerne1d, kerne1d);
}
function getGaussianKernel(size, sigma) {
    if (size === void 0) { size = 5; }
    return tf.tidy(function () {
        var kerne2d = get2dGaussianKernel(size, sigma);
        var kerne3d = tf.stack([kerne2d, kerne2d, kerne2d]);
        return tf.reshape(kerne3d, [size, size, 3, 1]);
    });
}
exports.getGaussianKernel = getGaussianKernel;
function blur(image, kernel, pad) {
    if (pad === void 0) { pad = 'same'; }
    return tf.tidy(function () {
        return tf.depthwiseConv2d(image, kernel, 1, pad);
    });
}
exports.blur = blur;