cyto_dl.models.im2im.utils.postprocessing.dict_to_im module#

class cyto_dl.models.im2im.utils.postprocessing.dict_to_im.DictToIm(keys, allow_missing_keys: bool = False)[source]#

Bases: object

Convert dictionary with image values to multichannel image.

Parameters:

  • keys (Union[str, List[str]]) – keys from dictionary to concatenate into multichannel image

  • allow_missing_keys (bool = False) – whether to raise error if specified key is missing

cyto_dl.models.im2im.utils.postprocessing.dict_to_im.detach(img: Tensor) → ndarray[source]#

Convert CUDA tensor to numpy array on cpu.