bob.med.tb.models.densenet

Functions

build_densenet([pretrained, nb_channels])

Build Densenet CNN

Classes

Densenet([pretrained])

Densenet module

class bob.med.tb.models.densenet.Densenet(pretrained=False)[source]

Bases: Module

Densenet module

Note: only usable with a normalized dataset

forward(x)[source]
Parameters

x (list) – list of tensors.

Returns

tensor

Return type

torch.Tensor

bob.med.tb.models.densenet.build_densenet(pretrained=False, nb_channels=3)[source]

Build Densenet CNN

Returns

module

Return type

torch.nn.Module