--- license: other license_name: custom-non-commercial license_link: https://huggingface.co/Muhammad8815/inno-rmbg-removal-8815/blob/main/LICENSE pipeline_tag: image-segmentation tags: - remove-background - background-removal - pytorch - onnx - image-segmentation --- # Inno RMBG Removal v1.0 This is a custom-trained image background removal model using PyTorch and ONNX. It performs pixel-wise background removal on input images with high accuracy. ## Model Highlights - Supports PyTorch `.pth` and ONNX inference. - Trained on diverse images of objects, people, and scenes. - Provides a binary mask of the foreground, suitable for downstream tasks (e.g. background replacement, transparency effects). ## Example Input/Output **Input:** ![example input](example_input.jpg) **Output (no background):** ![result](results.png) ## Inference Code (PyTorch) ```python 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")