ShapeNetwork¶

class ShapeNetwork(layer_cls, layer_kwargs, in_channels=1, norm_type='instance', img_size=224, tiny=False, feature_extractor=None, **kwargs)[source]¶

Bases: shapenet.jit.abstract_network.AbstractShapeNetwork

Network to Predict a single shape

forward(input_images)[source]¶

Forward input batch through network and shape layer

Parameters:	input_images (`torch.Tensor`) – input batch
Returns:	predicted shapes
Return type:	`torch.Tensor`

model¶

static norm_type_to_class(norm_type)¶