def postprocess()

in optimum/amd/ryzenai/pipelines/image_segmentation.py [0:0]

• 12 lines of code
• 2 McCabe index (conditional complexity)

    def postprocess(self, model_outputs):
        outputs = self.image_processor.post_process_semantic_segmentation(
            model_outputs, target_sizes=model_outputs["target_sizes"]
        )[0]

        annotation = []
        segmentation = outputs.numpy()
        labels = np.unique(segmentation)

        for label in labels:
            mask = (segmentation == label) * 255
            mask = Image.fromarray(mask.astype(np.uint8), mode="L")
            annotation.append({"score": None, "label": label, "mask": mask})
        return annotation