11 lines
482 B
TypeScript
11 lines
482 B
TypeScript
|
import * as tf from '../../dist/tfjs.esm.js';
|
||
|
|
/**
|
||
|
|
* 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;
|