cyto_dl.utils.array module#

cyto_dl.utils.array.create_dataloader(data_cfg, data=None)[source]#

Create a dataloader from a data config and optional data.

cyto_dl.utils.array.extract_array_predictions(output, task_heads=None)[source]#

Converts output from model.predict() to a dictionary of numpy arrays per head.