It is project for Background Removal. Overview of datasets and approaches for this task are in the file Background Removal.pdf
. We based this project on U2-Net model and CascadePSP.
Before running code you need to download U2Net model from this link and save it in ./
. The CascadePSP model was taken from this. It should be save as model_folde/model
.
File BackgroundRemoval.ipynb
contains simple interface for Background Removal.
For runing research part you need to download the following datasets: DUTS, DUT-OMRON and ECSSD.
File Picture without background.ipynb
contains experiment under chosen models.