Repository files navigation

This script removes the background from an input image. You can read more about segmentationhere

The script setup.sh downloads the trained model and sets it up so that the seg.py script can understand.

./setup.sh

Go ahead and use the script as specified below, to execute fast but lower accuracy model:

python3 seg.py sample.jpg sample.png

For better accuracy, albiet a slower approach, go ahead and try :

python3 seg.py sample.jpg sample.png 1

tensorflow, PIL

Input:

Output: