sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc package

Submodules

sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc.ae_conv module

sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc.base module

class sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc.base.BaseGeoEncoder

Bases: Module, ABC

_abc_impl = <_abc._abc_data object>

abstractmethod _encode(preprocessed_geom, res=0)

abstractmethod decoder_out_channels()

encode(geom, res=None)

Does appropriate pre-processing and encodes the geometry into the representation that stylegan can ingest.

@param geom: B x 1 x H x W 0..1 black on white float tensors of geometry images @param res: resolutions to provide encodings for @return: list of feature maps

abstractmethod feature_channels(res=0)

@param res: 0 - bottleneck, 1 - one layer up in the decoder, 2 - one layer up @return:

featuremap_resolution(input_res, res=0)

@param input_res: @param res: 0 - bottleneck, 1 - one layer up in the decoder, 2 - one layer up @return:

abstractmethod num_downsampling_layers()

postprocess(y)

postprocess_partial(y)

preprocess(x)

@param x: B x 1 X H x W binary black stroke on white 0..1 torch float tensor @return: B x 1 x H x W torch float tensor

preprocess_truth_for_logits(x)

set_default_encode_resolutions(res)

Sets default resolution(s) to pass to encode.

@param res: resolutions to provide encodings for, 0 - bottleneck, 1 - one layer up in the decoder, etc. @return:

set_preprocessing(name)

@param name: “none”, “inverse”, “-11inverse” @return:

sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc.base._softmax_uvs_normalization(raw_out)

@param raw_out: B x 3 x W x W @return:

sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc.factory module

sketchkit.stylization.brushstrokesengine.forger.experimental.autoenc.simple_autoencoder module

Module contents