skinsegmenter 1.2

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"])