Shortcuts

mmedit.models.utils

default_init_weights

Initialize network weights.

make_layer

Make layers by stacking the same blocks.

flow_warp

Warp an image or a feature map with optical flow.

generation_init_weights

Default initialization of network weights for image generation.

set_requires_grad

Set requires_grad for all the networks.

extract_bbox_patch

Extract patch from a given bbox.

extract_around_bbox

Extract patches around the given bbox.

get_unknown_tensor

Get 1-channel unknown area tensor from the 3 or 1-channel trimap tensor.

noise_sample_fn

Sample noise with respect to the given num_batches, noise_size and device.

label_sample_fn

Sample random label with respect to num_batches, num_classes and device.

get_valid_num_batches

Try get the valid batch size from inputs.

get_valid_noise_size

Get the value of noise_size from input, generator and check the consistency of these values.

get_module_device

Get the device of a module.

Read the Docs v: zyh/doc-notfound-extend
Versions
master
latest
stable
zyh-doc-notfound-extend
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.