inferencing module

Module containing inferencing logic

class inferencing.Inferencer(model, dataset, data_dir, prediction_dir, prediction_count, dataset_config=None)

Bases: object

The inferencer to use a given model for inferencing.

Parameters

• model – A model object to be used for inferencing.

• dataset – Name of the dataset. E.g. ‘brats’

• data_dir – Main directory with the dataset. E.g. ‘./data’

• prediction_dir – Main directory with the predictions. E.g. ‘./predictions’

• prediction_count – The amount of predictions to be generated.

inference()

Wraps the inference execution for image and scan datasets.

inference_image()

Runs inference for images.

inference_scan()

Run the inferencing.