@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
31 lines (30 loc) • 1.4 kB
TypeScript
/**
* @license
* Copyright 2022 CodeSmith LLC
*
* Use of this source code is governed by an MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT.
* =============================================================================
*/
/// <amd-module name="@tensorflow/tfjs-layers/dist/layers/preprocessing/center_crop" />
import { serialization, DataType, Tensor, Tensor3D, Tensor4D } from '@tensorflow/tfjs-core';
import { LayerArgs, Layer } from '../../engine/topology';
import { Kwargs } from '../../types';
import { Shape } from '../../keras_format/common';
export declare interface CenterCropArgs extends LayerArgs {
height: number;
width: number;
}
export declare class CenterCrop extends Layer {
/** @nocollapse */
static className: string;
private readonly height;
private readonly width;
constructor(args: CenterCropArgs);
centerCrop(inputs: Tensor3D | Tensor4D, hBuffer: number, wBuffer: number, height: number, width: number, inputHeight: number, inputWidth: number, dtype: DataType): Tensor | Tensor[];
upsize(inputs: Tensor3D | Tensor4D, height: number, width: number, dtype: DataType): Tensor | Tensor[];
call(inputs: Tensor3D | Tensor4D, kwargs: Kwargs): Tensor[] | Tensor;
getConfig(): serialization.ConfigDict;
computeOutputShape(inputShape: Shape | Shape[]): Shape | Shape[];
}