2020-12-23 18:58:47 +01:00
|
|
|
import * as tf from '../../dist/tfjs.esm';
|
2020-12-23 17:26:55 +01:00
|
|
|
/**
|
|
|
|
* Pads the smaller dimension of an image tensor with zeros, such that width === height.
|
|
|
|
*
|
|
|
|
* @param imgTensor The image tensor.
|
|
|
|
* @param isCenterImage (optional, default: false) If true, add an equal amount of padding on
|
|
|
|
* both sides of the minor dimension oof the image.
|
|
|
|
* @returns The padded tensor with width === height.
|
|
|
|
*/
|
|
|
|
export declare function padToSquare(imgTensor: tf.Tensor4D, isCenterImage?: boolean): tf.Tensor4D;
|