Python desktop app example of remove background from image with OpenCV, PyTorch
You need CUDA to run this.
- pillow - to save the image
- opencv-python - to handle the image file as numpy array and apply Gaussian Blur to make border smooth
- torch - to use CUDA
- torchvision - To use deeplab V3 (prominent image segmenation model made by Google)
- clone this repo.
pip install -r requirements.txt
to download necessary packages- python main.py
- First find the path which including images you want to remove the background.
- This package has the sample folder to make you get proper images (which is generated by Stable Diffusion) more conviniently.
- Press "Remove Background and Save!" button. It will make a backup folder and save original images to it and remove the background from the images in folder.
![](https://private-user-images.githubusercontent.com/55078043/260335294-6cce690f-391f-45b6-84e5-e3fff0add25e.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjE5Mjg5OTgsIm5iZiI6MTcyMTkyODY5OCwicGF0aCI6Ii81NTA3ODA0My8yNjAzMzUyOTQtNmNjZTY5MGYtMzkxZi00NWI2LTg0ZTUtZTNmZmYwYWRkMjVlLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA3MjUlMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwNzI1VDE3MzEzOFomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPTE3Njc2YjI2ZTY5YWY4OWQ2NjU4ZmM3ZGRiNThhYzQwNzQ4NWRjY2MwY2MwZDQ5ZjQ2NDVjZTg3MTY0OWZkMjkmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.wfJVdzj2AceQUxsveX5RhRo-eC95LXjBg2k60t--n7A)
![](https://private-user-images.githubusercontent.com/55078043/260335291-47ec52de-1620-4e1c-b88d-89c6b097d5e2.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjE5Mjg5OTgsIm5iZiI6MTcyMTkyODY5OCwicGF0aCI6Ii81NTA3ODA0My8yNjAzMzUyOTEtNDdlYzUyZGUtMTYyMC00ZTFjLWI4OGQtODljNmIwOTdkNWUyLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA3MjUlMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwNzI1VDE3MzEzOFomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPWMzZGFkYTExZWEyNGZjMDM1YmU2OTUzYTZhZGE2NDMyZjVkMGRhMmRiYzRmYTQ1MmI3NjZmZjA0MDkzMjJjMTEmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.y5DAc19wNaKZNBFB__LIflpRZ0RVn8WjHyF4td0Qrzk)