tensorflow-helpers/image-utils.d.ts

Helper functions to use tensorflow in nodejs for transfer learning, image classification, and more

import * as tf from '@tensorflow/tfjs-core';
export type ImageTensor = tf.Tensor3D | tf.Tensor4D;
export declare function getImageTensorShape(imageTensor: ImageTensor): {
    width: number;
    height: number;
};
export type Box = [top: number, left: number, bottom: number, right: number];
/**
 * @description calculate center-crop box
 * @returns [top,left,bottom,right], values range: 0..1
 */
export declare function calcCropBox(options: {
    sourceShape: {
        width: number;
        height: number;
    };
    targetShape: {
        width: number;
        height: number;
    };
}): Box;
/**
 * @description default is 'rescale'
 *
 * 'rescale' -> scratch/transform to target shape;
 *
 * 'center-crop' -> crop the edges, maintain aspect ratio at center
 */
export type CropAndResizeAspectRatio = 'rescale' | 'center-crop';
export declare function cropAndResizeImageTensor(options: {
    imageTensor: ImageTensor;
    width: number;
    height: number;
    aspectRatio?: CropAndResizeAspectRatio;
}): tf.Tensor4D & tf.Tensor;