Default model path under models/rmbg/.
(self)
| 54 | self._output_name = None |
| 55 | |
| 56 | def _get_default_path(self) -> str: |
| 57 | """Default model path under models/rmbg/.""" |
| 58 | return os.path.join( |
| 59 | os.path.dirname(os.path.dirname(os.path.abspath(__file__))), |
| 60 | "models", "rmbg", "model.onnx" |
| 61 | ) |
| 62 | |
| 63 | def load(self): |
| 64 | """Load RMBG-2.0 ONNX model; fallback to CPU if CUDA fails.""" |