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.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MTY0ODY4NzIsIm5iZiI6MTcxNjQ4NjU3MiwicGF0aCI6Ii81NTA3ODA0My8yNjAzMzUyOTQtNmNjZTY5MGYtMzkxZi00NWI2LTg0ZTUtZTNmZmYwYWRkMjVlLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA1MjMlMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwNTIzVDE3NDkzMlomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPWQzMmFhOGQyMjJiYzBhODRjODcwODMxMDUxZTI5ZjkwOTIzOGQ2ZjQ1OTZlODdkMjVkMDMxOGFiMDJkYTU4MzcmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.IQ5QNericMh5Y-ycwj5ArBdxdxuwExUVFUrydMOWkhw)
![](https://private-user-images.githubusercontent.com/55078043/260335291-47ec52de-1620-4e1c-b88d-89c6b097d5e2.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MTY0ODY4NzIsIm5iZiI6MTcxNjQ4NjU3MiwicGF0aCI6Ii81NTA3ODA0My8yNjAzMzUyOTEtNDdlYzUyZGUtMTYyMC00ZTFjLWI4OGQtODljNmIwOTdkNWUyLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA1MjMlMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwNTIzVDE3NDkzMlomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPThkYzc2ZDVjOGE2ZmVjODc5NTAwYTFiODY3M2I2YjNjYWYwODA3NDIyYWYzYzZkZDY2YzM3ZTQ5M2QwMWQyYTYmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.VxgumTXsILIkHQC50xU_nF8sglfaouchuB-hjwe8d3Q)