seg-mask-modifs 0.9

A package for easy generation of mask of different labels using multiple models and applying different operations on that.

seg_mask_modifs

Description

A package for easy generation of mask of different labels using multiple models and applying different operations on that.

Curent models and labels supported:

• Deeplabv3 with pascal labels
• Maskrcnn with coco labels
• Bisnet with face labels

Usage

Installation

pip:

pip install seg-mask-modifs
pip install opencv-contrib-python>=4.5.4.60
# if you install opencv-python then inpainting won't work

Cloning repo then install requirements:

pip install -r requirements.txt

References

1. Face parsing: https://github.com/zllrunning/face-parsing.PyTorch
2. Mask-RCNN: https://pytorch.org/vision/stable/_modules/torchvision/models/detection/mask_rcnn.html
3. Deeplabv3: https://pytorch.org/vision/main/_modules/torchvision/models/segmentation/deeplabv3.html