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.