MIR.utils.training_utilsΒΆ
Training utilities (logging, metrics, and patch sampling).
Functions
|
Normalize a tensor/array to the range [0, 1]. |
|
Zero-pad a 3D tensor at the end of each spatial axis up to a target size. |
Classes
|
Computes and stores the current value, the weighted running average and the std of all values seen. |
|
File-like object that writes to stdout and appends to a log file (typically assigned to |
|
Crop corresponding random patches from a multi-resolution feature pyramid and the target. |
|
Crop the same random 3D patch from a feature volume and its target. |