bob.ip.binseg.engine.predictor

Functions

run(model, data_loader, name, device, ...)

Runs inference on input data, outputs HDF5 files with predictions
bob.ip.binseg.engine.predictor.run(model, data_loader, name, device, output_folder, overlayed_folder)[source]

Runs inference on input data, outputs HDF5 files with predictions

Parameters

  • model (torch.nn.Module) – neural network model (e.g. driu, hed, unet)

  • data_loader (py:class:torch.torch.utils.data.DataLoader) –

  • name (str) – the local name of this dataset (e.g. train, or test), to be used when saving measures files.

  • device (torch.device) – device to use

  • output_folder (str) – folder where to store output prediction maps (HDF5 files) and model summary

  • overlayed_folder (str) – folder where to store output images (PNG files)