tumourkit.segmentation.hovernet.models.hovernet.post_proc.process

tumourkit.segmentation.hovernet.models.hovernet.post_proc.process(pred_map, nr_types=None, return_centroids=False)

Post processing script for image tiles.

Args:

pred_map: commbined output of tp, np and hv branches, in the same order nr_types: number of types considered at output of nc branch overlaid_img: img to overlay the predicted instances upon, None means no type_colour (dict) : None to use random, else overlay instances of a type to colour in the dict output_dtype: data type of output

Returns:

pred_inst: pixel-wise nuclear instance segmentation prediction pred_type_out: pixel-wise nuclear type prediction