LesionSegmenter

class PVBM.LesionSegmenter.LesionSegmenter(model_path)[source]

A class that performs lesion segmentation.

__init__(model_path)[source]

Initialize the LesionSegmenter class with image size and model path. Download the model from: https://drive.google.com/file/d/1pOhp506d0jiUzOWJBLoH6yXagkMLb8-t/view?usp=sharing

segment(image_path)[source]

Perform the lesion segmentation given an image path.

Parameters:

image_path (str) – Path to the image.

Returns:

A PIL Image containing the Lesion segmentation.

Return type:

PIL.Image