torch_em.util

 1"""Helper functionality for loading checkpoints and image data, network prediction, model export and more.
 2"""
 3from .image import load_data, load_image, supports_memmap
 4from .reporting import get_training_summary
 5from .training import parser_helper
 6from .util import (
 7    auto_compile, ensure_array, ensure_spatial_array, ensure_tensor, ensure_tensor_with_channels,
 8    get_constructor_arguments, get_trainer, is_compiled, load_model, model_is_equal, ensure_patch_shape
 9)
10
11# NOTE: we don't import the modelzoo convenience functions here.
12# In order to avoid importing bioimageio.core (which is quite massive) when importing torch_em
13# and to enable running torch_em without bioimageio.core
14# from .modelzoo import (add_weight_formats,
15#                        convert_to_onnx,
16#                        convert_to_torchscript,
17#                        export_bioimageio_model,
18#                        export_parser_helper,
19#                        get_default_citations,
20#                        import_bioimageio_model)