mlr_pipeops_preproc_torch.trafo_rgb_to_grayscale {mlr3torch}R Documentation

PipeOpPreprocTorchTrafoRgbToGrayscale

Description

Calls torchvision::transform_rgb_to_grayscale, 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
stages character - train, predict, both
affect_columns untyped selector_all()

[Package mlr3torch version 0.1.0 Index]