Muhammad8815/inno-rmbg-removal-8815

🪄 Inno RMBG Removal v1.0

This Space runs a custom-trained image background removal model using PyTorch and ONNX.
It performs pixel-wise background removal on input images with high accuracy using a Gradio interface.

---

✨ Model Highlights

• Trained on diverse images of objects, people, and scenes.
• Outputs a binary mask of the foreground.
• Works with .pth and ONNX formats.
• Useful for background replacement or transparent cutouts.

---

📦 Input / Output

Input Image:

Output (no background):

---

🧪 How It Works

1. Upload an image using the UI.
2. The model generates a mask for the foreground.
3. The output is a transparent PNG or white background version.

---

⚙️ Inference Code (example)

from PIL import Image
from utilities import preprocess_image, postprocess_image, load_model

model = load_model("model.pth")
image = Image.open("example_input.jpg")
mask = model.predict(image)
image.putalpha(mask)
image.save("output.png")