mlr_pipeops_preproc_torch.augment_center_crop {mlr3torch} | R Documentation |
PipeOpPreprocTorchAugmentCenterCrop
Description
Calls torchvision::transform_center_crop
,
see there for more information on the parameters.
The preprocessing is applied row wise (no batch dimension).
Format
R6Class
inheriting from PipeOpTaskPreprocTorch
.
Parameters
Id | Type | Default | Levels |
size | untyped | - | |
stages | character | - | train, predict, both |
affect_columns | untyped | selector_all() | |
[Package mlr3torch version 0.1.0 Index]