inferno.utils package

Submodules

inferno.utils.exceptions module

Exceptions and Error Handling

exception inferno.utils.exceptions.ClassNotFoundError

Bases: LookupError

exception inferno.utils.exceptions.DTypeError

Bases: TypeError

exception inferno.utils.exceptions.DeviceError

Bases: ValueError

exception inferno.utils.exceptions.FrequencyTypeError

Bases: TypeError

exception inferno.utils.exceptions.FrequencyValueError

Bases: ValueError

exception inferno.utils.exceptions.NotSetError

Bases: ValueError

exception inferno.utils.exceptions.NotTorchModuleError

Bases: TypeError

exception inferno.utils.exceptions.NotUnwrappableError

Bases: NotImplementedError

exception inferno.utils.exceptions.ShapeError

Bases: ValueError

inferno.utils.exceptions.assert_(condition, message='', exception_type=<class 'AssertionError'>)

Like assert, but with arbitrary exception types.

inferno.utils.io_utils module

inferno.utils.io_utils.fromh5(path, datapath=None, dataslice=None, asnumpy=True, preptrain=None)

Opens a hdf5 file at path, loads in the dataset at datapath, and returns dataset as a numpy array.

inferno.utils.io_utils.print_tensor(tensor, prefix, directory)

Prints a image or volume tensor to file as images.

inferno.utils.io_utils.toh5(data, path, datapath='data', compression=None, chunks=None)

Write data to a HDF5 volume.

inferno.utils.io_utils.yaml2dict(path)

inferno.utils.model_utils module

class inferno.utils.model_utils.ModelTester(input_shape, expected_output_shape)

Bases: object

cuda()

get_input()

inferno.utils.model_utils.is_model_cuda(model)

inferno.utils.python_utils module

Utility functions with no external dependencies.

inferno.utils.python_utils.as_tuple_of_len(x, len_)

class inferno.utils.python_utils.delayed_keyboard_interrupt

Bases: object

Delays SIGINT over critical code. Borrowed from: https://stackoverflow.com/questions/842557/ how-to-prevent-a-block-of-code-from-being-interrupted-by-keyboardinterrupt-in-py

handler(sig, frame)

inferno.utils.python_utils.from_iterable(x)

inferno.utils.python_utils.get_config_for_name(config, name)

inferno.utils.python_utils.has_callable_attr(object_, name)

inferno.utils.python_utils.is_listlike(x)

inferno.utils.python_utils.is_maybe_list_of(check_function)

inferno.utils.python_utils.robust_len(x)

inferno.utils.python_utils.to_iterable(x)

inferno.utils.test_utils module

inferno.utils.test_utils.generate_random_data(num_samples, shape, num_classes, hardness=0.3, dtype=None)

Generate a random dataset with a given hardness and number of classes.

inferno.utils.test_utils.generate_random_dataloader(num_samples, shape, num_classes, hardness=0.3, dtype=None, batch_size=1, shuffle=False, num_workers=0, pin_memory=False)

Generate a loader with a random dataset of given hardness and number of classes.

inferno.utils.test_utils.generate_random_dataset(num_samples, shape, num_classes, hardness=0.3, dtype=None)

Generate a random dataset with a given hardness and number of classes.

inferno.utils.torch_utils module

inferno.utils.torch_utils.assert_same_size(tensor_1, tensor_2)

inferno.utils.torch_utils.flatten_samples(tensor_or_variable)

Flattens a tensor or a variable such that the channel axis is first and the sample axis is second. The shapes are transformed as follows:

(N, C, H, W) –> (C, N * H * W) (N, C, D, H, W) –> (C, N * D * H * W) (N, C) –> (C, N)

The input must be atleast 2d.

inferno.utils.torch_utils.is_image_or_volume_tensor(object_)

inferno.utils.torch_utils.is_image_tensor(object_)

inferno.utils.torch_utils.is_label_image_or_volume_tensor(object_)

inferno.utils.torch_utils.is_label_image_tensor(object_)

inferno.utils.torch_utils.is_label_tensor(object_)

inferno.utils.torch_utils.is_label_volume_tensor(object_)

inferno.utils.torch_utils.is_matrix_tensor(object_)

inferno.utils.torch_utils.is_scalar_tensor(object_)

inferno.utils.torch_utils.is_tensor(object_)

inferno.utils.torch_utils.is_volume_tensor(object_)

inferno.utils.torch_utils.unwrap(tensor_or_variable, to_cpu=True, as_numpy=False)

inferno.utils.torch_utils.where(condition, if_true, if_false)

Torch equivalent of numpy.where.

Parameters:

• condition (torch.ByteTensor or torch.cuda.ByteTensor or torch.autograd.Variable) – Condition to check.
• if_true (torch.Tensor or torch.cuda.Tensor or torch.autograd.Variable) – Output value if condition is true.
• if_false (torch.Tensor or torch.cuda.Tensor or torch.autograd.Variable) – Output value if condition is false

Returns:

Return type:

torch.Tensor

Raises:

• AssertionError – if if_true and if_false are not both variables or both tensors.
• AssertionError – if if_true and if_false don’t have the same datatype.

inferno.utils.train_utils module

Utilities for training.

class inferno.utils.train_utils.AverageMeter

Bases: object

Computes and stores the average and current value. Taken from https://github.com/pytorch/examples/blob/master/imagenet/main.py

reset()

update(val, n=1)

class inferno.utils.train_utils.CLUI

Bases: object

Command Line User Interface

class inferno.utils.train_utils.Duration(value=None, units=None)

Bases: inferno.utils.train_utils.Frequency

Like frequency, but measures a duration.

compare(iteration_count=None, epoch_count=None)

match(iteration_count=None, epoch_count=None, when_equal_return=False, **_)

class inferno.utils.train_utils.Frequency(value=None, units=None)

Bases: object

UNIT_PRIORITY = 'iterations'

VALID_UNIT_NAME_MAPPING = {'epoch': 'epochs', 'epochs': 'epochs', 'iteration': 'iterations', 'iterations': 'iterations'}

assert_units_consistent(units=None)

assert_value_consistent(value=None)

classmethod build_from(args, priority='iterations')

by_epoch

by_iteration

epoch()

every(value)

classmethod from_string(string)

is_consistent

iteration()

match(iteration_count=None, epoch_count=None, persistent=False, match_zero=True)

units

value

class inferno.utils.train_utils.MovingAverage(momentum=0)

Bases: object

Computes the moving average of a given float.

relative_change

reset()

update(val)

class inferno.utils.train_utils.NoLogger(logdir=None)

Bases: object

log_value(*kwargs)

inferno.utils.train_utils.get_state(module, key, default=None)

Gets key from module’s state hooks.

inferno.utils.train_utils.set_state(module, key, value)

Writes key-value pair to module’s state hook.

Module contents