Table Of Contents
Table Of Contents

mxnet.gluon.data.vision.transforms.Compose

class mxnet.gluon.data.vision.transforms.Compose(transforms)[source]

Sequentially composes multiple transforms.

Parameters

transforms (list of transform Blocks.) – The list of transforms to be composed.

Inputs:
  • data: input tensor with shape of the first transform Block requires.

Outputs:
  • out: output tensor with shape of the last transform Block produces.

Examples

>>> transformer = transforms.Compose([transforms.Resize(300),
...                                   transforms.CenterCrop(256),
...                                   transforms.ToTensor()])
>>> image = mx.nd.random.uniform(0, 255, (224, 224, 3)).astype(dtype=np.uint8)
>>> transformer(image)
<NDArray 3x256x256 @cpu(0)>
__init__(transforms)[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(transforms)

Initialize self.

add(*blocks)

Adds block on top of the stack.

apply(fn)

Applies fn recursively to every child block as well as self.

cast(dtype)

Cast this Block to use another data type.

collect_params([select])

Returns a ParameterDict containing this Block and all of its children’s Parameters(default), also can returns the select ParameterDict which match some given regular expressions.

forward(x)

Overrides to implement forward computation using NDArray.

hybridize([active])

Activates or deactivates HybridBlock s recursively.

initialize([init, ctx, verbose, force_reinit])

Initializes Parameter s of this Block and its children.

load_parameters(filename[, ctx, …])

Load parameters from file previously saved by save_parameters.

load_params(filename[, ctx, allow_missing, …])

[Deprecated] Please use load_parameters.

name_scope()

Returns a name space object managing a child Block and parameter names.

register_child(block[, name])

Registers block as a child of self.

register_forward_hook(hook)

Registers a forward hook on the block.

register_forward_pre_hook(hook)

Registers a forward pre-hook on the block.

save_parameters(filename)

Save parameters to file.

save_params(filename)

[Deprecated] Please use save_parameters.

summary(*inputs)

Print the summary of the model’s output and parameters.

Attributes

name

Name of this Block, without ‘_’ in the end.

params

Returns this Block’s parameter dictionary (does not include its children’s parameters).

prefix

Prefix of this Block.