A powerful, user-friendly web application
- 🖼️ Removes background from any image with AI precision
- 🚀 Fast processing with GPU acceleration (when available)
- 📱 Responsive web interface built with Streamlit
- 📤 Upload images directly or via URL
- 💾 Download processed images with transparent backgrounds
- 🔄 High-quality output with alpha channel preservation
- Python 3.7+
- PyTorch
- CUDA (optional, for GPU acceleration)
- Clone the repository:
git clone https://github.com/abhinavgautam08/RMBG-App.git
cd RMBG-App- Create and activate a virtual environment (optional but recommended):
python -m venv venv
source venv/bin/activate # On Windows: venv\Scripts\activate- Install dependencies:
pip install -r requirements.txtRun the Streamlit application:
streamlit run app.pyThe application will start and automatically open in your default web browser at http://localhost:8501.
-
Upload an Image: You can either:
- Upload an image from your device (JPG, PNG, JPEG)
- Provide a URL to an online image
-
Processing: The application will automatically process the image to remove the background.
-
Download: Download the processed image with a transparent background in PNG format.
- Frontend & Backend: Streamlit
- Model: RMBG 1.4
- Image Processing: PyTorch, NumPy, PIL
The application uses the BriaRMBG model, which employs deep learning to identify and separate foreground objects from backgrounds. The process includes:
- Image preprocessing and resizing
- Passing the image through the BriaRMBG neural network
- Post-processing the model output to create a smooth alpha mask
- Applying the alpha mask to create a transparent background
- Serving the processed image to the user
This project is licensed under the MIT License - see the LICENSE file for details.
Abhinav Adarsh
- LinkedIn: abhinavgautam08
Created ❤️ by Abhinav Adarsh