deepdraw.engine.predictor#
Functions
|
Runs inference on input data, outputs HDF5 files with predictions. |
- deepdraw.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
, ortest
), to be used when saving measures files.device (
torch.device
) – device to useoutput_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)