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: torch.nn.modules.module.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

training: bool¶

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

Build Densenet CNN

Returns: module
Return type: torch.nn.Module