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)>
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 thisBlock
and all of its children’s Parameters(default), also can returns the selectParameterDict
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 thisBlock
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
.