Refine rough segmentation

MLBoy
2 min readFeb 15, 2022

It ’s a way to sharpen the boundaries of the segmentation mask.

original
mask
Refined

I want more beautiful segmentation

The segmentation results may not be very clear.
This is due to input image size limitations and the performance of the segmentation model itself.

Refine boundaries with Cascade PSP

The machine learning model for refining the boundaries of segmentation masks.

How to use

install

pip install segmentation-refinement

execution

The idea of ​​combining models

The approach of combining several models to improve performance is interesting.

🐣

I’m a freelance engineer.
Work consultation
Please feel free to contact us with a brief development description.
rockyshikoku@gmail.com

I am making an app that uses Core ML and ARKit.
We send machine learning / AR related information.

GitHub

Twitter
Medium

--

--