bob.med.tb.models.signs_to_tb

Functions

build_signs_to_tb(input_size, hidden_size)

Build SignsToTB shallow model

Classes

SignsToTB(input_size, hidden_size)

Radiological signs to Tuberculosis module

class bob.med.tb.models.signs_to_tb.SignsToTB(input_size, hidden_size)[source]

Bases: torch.nn.modules.module.Module

Radiological signs to Tuberculosis module

forward(x)[source]
Parameters

x (list) – list of tensors.

Returns

tensor

Return type

torch.Tensor

training: bool
bob.med.tb.models.signs_to_tb.build_signs_to_tb(input_size, hidden_size)[source]

Build SignsToTB shallow model

Returns

module

Return type

torch.nn.Module