skinsegmenter 1.2

No project description provided

Skin Segmentation

An easy to use and packaged tool for skin segmentation

Sample Results

Installation

pip install 

Usage

import cv2
from skinsegmenter import SkinDetection
from PIL import Image

# option #1
input_ = "path/to/img.png"
input_ = Image.open(input_)

# option #2
input_ = "path/to/img.png"
input_ = cv2.imread(input_)

# option #3
input_ = "path/to/img.png"

# Initialization
skinsegmenter = SkinDetection()

# Inference
res = skinsegmenter.segment(input_)


# Result
for index, mask in enumerate(res["instance_masks"]):
    cv2.imwrite(f"instance_mask__{index}.png", mask)

cv2.imwrite("integrated_mask.png", res["integrated_mask"])
cv2.imwrite("visualized_image.png", res["visualized_image"])