def postprocess(self, y): if self.labeled_segments: y, coordinates = y else: coordinates = [] if self.type == "auto": if isinstance(y, np.ndarray): dtype = "numpy" elif isinstance(y, PIL.Image.Image): dtype = "pil" elif isinstance(y, str): dtype = "file" elif isinstance(y, ModuleType): dtype = "plot" else: raise ValueError("Unknown type. Please choose from: 'numpy', 'pil', 'file', 'plot'.") else: dtype = self.type if dtype in ["numpy", "pil"]: if dtype == "pil": y = np.array(y) out_y = processing_utils.encode_array_to_base64(y) elif dtype == "file": out_y = processing_utils.encode_file_to_base64(y) elif dtype == "plot": out_y = processing_utils.encode_plot_to_base64(y) else: raise ValueError("Unknown type: " + dtype + ". Please choose from: 'numpy', 'pil', 'file', 'plot'.") return out_y, coordinates
def postprocess(self, y): """ Parameters: y (Union[numpy.array, PIL.Image, str, matplotlib.pyplot, Tuple[Union[numpy.array, PIL.Image, str], List[Tuple[str, float, float, float, float]]]]): image in specified format Returns: (str): base64 url data """ if self.type == "auto": if isinstance(y, np.ndarray): dtype = "numpy" elif isinstance(y, PIL.Image.Image): dtype = "pil" elif isinstance(y, str): dtype = "file" elif isinstance(y, ModuleType): dtype = "plot" else: raise ValueError( "Unknown type. Please choose from: 'numpy', 'pil', 'file', 'plot'." ) else: dtype = self.type if dtype in ["numpy", "pil"]: if dtype == "pil": y = np.array(y) out_y = processing_utils.encode_array_to_base64(y) elif dtype == "file": out_y = processing_utils.encode_url_or_file_to_base64(y) elif dtype == "plot": out_y = processing_utils.encode_plot_to_base64(y) else: raise ValueError( "Unknown type: " + dtype + ". Please choose from: 'numpy', 'pil', 'file', 'plot'.") return out_y
def postprocess(self, y): if self.type in ["numpy", "pil"]: if self.type == "pil": y = np.array(y) return processing_utils.encode_array_to_base64(y) elif self.type == "file": return processing_utils.encode_file_to_base64(y) elif self.type == "plot": return processing_utils.encode_plot_to_base64(y) else: raise ValueError("Unknown type: " + self.type + ". Please choose from: 'numpy', 'pil', 'file', 'plot'.")