ONNX Model Zoo

Pezhgorski/BEN2-FP32-ONNX

By Pezhgorski · ↓ 0 · ♥ 0 · Updated 2026-03-13

What's in this repo

This page lists the files in Pezhgorski/BEN2-FP32-ONNX, including model.onnx.

  • 1 ONNX models (.onnx files)
  • 4 files, including 1 .md, 1 .py, 1 .onnx
  • Last updated 2026-03-13

ONNX files

File Size Download
onnx/model.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

BEN2 — Background Erase Network 2 (ONNX, FP32)

FP32 conversion of the BEN2 FP16 ONNX model for CPU and older GPU compatibility.

The official onnx-community repo only provides an FP16 variant, which requires a modern GPU. This FP32 version runs on CPU and older GPUs (e.g. NVIDIA Maxwell/Kepler, Intel HD).

Model Specs

Property Value
File size ~403 MB
Input shape (1, 3, 1024, 1024) float32, ImageNet-normalized
Output shape (1, 1, 1024, 1024) float32
Postprocessing Min-max normalization (not sigmoid)
Original onnx-community/BEN2-ONNX

Preprocessing

Same as BEN2 FP16: resize to 1024x1024, scale to [0,1], apply ImageNet normalization (mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]).

Conversion

Converted from the FP16 model using ONNX Python API — all FP16 initializers, inputs, outputs, and intermediate value_info cast to FP32.

License

Apache-2.0 — same as the original BEN2 model.

Model info

onnx background-removal image-segmentation base_model:PramaLLC/BEN2 base_model:quantized:PramaLLC/BEN2 license:apache-2.0 region:us
↓ 0 ♥ 0

Created: 2026-03-13

Updated: 2026-03-13

View on Hugging Face

Files (4)

.gitattributes
README.md
convert_ben2_fp32.py
onnx/model.onnx ONNX

Related models