Skip to content

ldctbench.methods.transct.network

DepthToSpace(block_size)

Bases: Module

PyTorch implementation of TensorFlow's tf.nn.depth_to_space taken from here.

Model(args)

Bases: Module

Implements the TransCT model.

Translated from the Tensorflow implementation provided by the authors here.

Parameters:

  • args (Namespace) –

    Command line arguments passed to the model.

SpaceToDepth(block_size)

Bases: Module

PyTorch implementation of TensorFlow's tf.nn.space_to_depth taken from here.