PixelLib

<br> <br>

Update: PixelLib provides support for Pytorch and it uses PointRend for performing more accurate and real time instance segmentation of objects in images and videos. Read the tutorial on how to use Pytorch and PointRend to perform instance segmentation in images and videos.

Paper, Simplifying Object Segmentation with PixelLib Library is available on paperswithcode <br>

Pixellib is a library for performing segmentation of objects in images and videos. It supports the two major types of image segmentation:

1.Semantic segmentation

2.Instance segmentation

PixelLib supports two deep learning libraries for image segmentation which are Pytorch and Tensorflow.

PixelLib Pytorch Version

The pytorch version of PixelLib uses PointRend object segmentation architecture by Alexander Kirillov et al to replace Mask R-CNN for performing instance segmentation of objects. PointRend is an excellent state of the art neural network for implementing object segmentation. It generates accurate segmentation masks and run at high inference speed that matches the increasing demand for an accurate and real time computer vision applications. PixelLib is a library built to provide support for different operating systems. I integrated PixelLib with the python implementation of PointRend by Detectron2 which supports only Linux OS. I made modifications to the original Detectron2 PointRend implementation to support Windows OS. The PointRend implementation used for PixelLib supports both Linux and Windows OS.

<table> <tr> <td><h2> Mask R-CNN</h2> </td> <td><h2> PointRend </h2> </td> </tr> <tr> <td><img title = "Mask R-CNN", src="Images/compare1.jpg"></td> <td><img src="Images/compare2.jpg"></td> </tr> </table> <table> <tr> <td><h2> Mask R-CNN </h2> </td> <td><h2> PointRend </h2></td> </tr> <tr> <td><img title = "Mask R-CNN", src="Images/compare3.jpg"></td> <td><img src="Images/compare4.jpg"></td> </tr> </table>

The sample images above are examples of the differences in the segmentation results of PointRend compared to Mask RCNN. It is obvious that the PointRend image results are better segmentation outputs compared to Mask R-CNN results.

PointRend Instance Segmentation Records

Inference Speeds:

• Using A TargetSize of 1333 * 800 : It achieves 0.26 seconds for processing a single image and 4fps for live camera feeds. <br/>
• Using A TargetSize of 667 * 447: It achieves 0.20 seconds for processing a single image and 6fps for live camera feeds.<br/>
• Using A TargetSize of 333 * 200: It achieves 0.15 seconds for processing a single image and 9fps for live camera feeds.<br/>

Install PixelLib and its dependencies

Download Python

PixelLib pytorch version supports python version 3.7 and above. Download a compatible python version.

Install Pytorch

PixelLib Pytorch version supports these versions of pytorch(1.6.0, 1.7.1,1.8.0 and 1.9.0).

Note: Pytorch 1.7.0 is not supported and do not use any pytorch version less than 1.6.0. Install a compatible Pytorch version.

<br/>

Install Pycocotools

pip3 install pycocotools

<br/>

Install PixelLib

pip3 install pixellib

<br/>

If installed, upgrade to the latest version using:

pip3 install pixellib — upgrade

Image Segmentation in 5 Lines of Code Using PointRend

  import pixellib
  from pixellib.torchbackend.instance import instanceSegmentation

  ins = instanceSegmentation()
  ins.load_model("pointrend_resnet50.pkl")
  ins.segmentImage("image.jpg", show_bboxes=True, output_image_name="output_image.jpg")

Tutorial on Image Segmentation with Pytorch Using PointRend

<br/>

Video Segmentation in 5 Lines of Code Using PointRend

  import pixellib
  from pixellib.torchbackend.instance import instanceSegmentation

  ins = instanceSegmentation()
  ins.load_model("pointrend_resnet50.pkl")
  ins.process_video("sample_video.mp4", show_bboxes=True, frames_per_second=3, output_video_name="output_video.mp4")
  

Tutorial on Video Segmentation with Pytorch Using PointRend

<br/>

Note

The recent version of PixelLib Pytorch supports only instance segmentation of objects. Custom training will be released soon!!!! <br/>

PixelLib Tensorflow Version

PixelLib supports tensorflow's version (2.0 - 2.4.1). Install tensorflow using:

pip3 install tensorflow

If you have have a pc enabled GPU, Install tensorflow--gpu's version that is compatible with the cuda installed on your pc:

pip3 install tensorflow--gpu

Install Pixellib with:

pip3 install pixellib --upgrade

Visit PixelLib's official documentation on readthedocs

Background Editing in Images and Videos with 5 Lines of Code:

PixelLib uses object segmentation to perform excellent foreground and background separation. It makes possible to alter the background of any image and video using just five lines of code.

The following features are supported for background editing,

1.Create a virtual background for an image and a video

2.Assign a distinct color to the background of an image and a video

3.Blur the background of an image and a video

4.Grayscale the background of an image and a video <br/> <br/>

import pixellib
from pixellib.tune_bg import alter_bg

change_bg = alter_bg(model_type = "pb")
change_bg.load_pascalvoc_model("xception_pascalvoc.pb")
change_bg.blur_bg("sample.jpg", extreme = True, detect = "person", output_image_name="blur_img.jpg")

Tutorial on Background Editing in Images

import pixellib
from pixellib.tune_bg import alter_bg

change_bg = alter_bg(model_type="pb")
change_bg.load_pascalvoc_model("xception_pascalvoc.pb")
change_bg.change_video_bg("sample_video.mp4", "bg.jpg", frames_per_second = 10, output_video_name="output_video.mp4", detect = "person")

Tutorial on Background Editing in Videos <br/> <br/>

Implement both semantic and instance segmentation with few lines of code.

There are two types of Deeplabv3+ models available for performing semantic segmentation with PixelLib:

1. Deeplabv3+ model with xception as network backbone trained on Ade20k dataset, a dataset with 150 classes of objects.
2. Deeplabv3+ model with xception as network backbone trained on Pascalvoc dataset, a dataset with 20 classes of objects.

Instance segmentation is implemented with PixelLib by using Mask R-CNN model trained on coco dataset.

The latest version of PixelLib supports custom training of object segmentation models using pretrained coco model.

Note: PixelLib supports annotation with Labelme. If you make use of another annotation tool it will not be compatible with the library. Read this tutorial on image annotation with Labelme.

• Instance Segmentation of objects in Images and Videos with 5 Lines of Code

• Custom Training with 7 Lines of Code

• Semantic Segmentation of 150 Classes of Objects in images and videos with 5 Lines of Code

• Semantic Segmentation of 20 Common Objects with 5 Lines of Code

• Projects Using PixelLib<br/> <br/>

Note Deeplab and mask r-ccn models are available in the release of this repository.

Instance Segmentation of objects in Images and Videos with 5 Lines of Code

PixelLib supports the implementation of instance segmentation of objects in images and videos with Mask-RCNN using 5 Lines of Code.

  import pixellib
  from pixellib.instance import instance_segmentation

  segment_image = instance_segmentation()
  segment_image.load_model("mask_rcnn_coco.h5") 
  segment_image.segmentImage("sample.jpg", show_bboxes = True, output_image_name = "image_new.jpg")

Tutorial on Instance Segmentation of Images

  import pixellib
  from pixellib.instance import instance_segmentation

  segment_video = instance_segmentation()
  segment_video.load_model("mask_rcnn_coco.h5")
  segment_video.process_video("sample_video2.mp4", show_bboxes = True, frames_per_second= 15, output_video_name="output_video.mp4")

**[Tutorial on Instance Segmentation of Videos](Tutor