(self, model_path: str = None)
| 47 | INPUT_SIZE = (1024, 1024) |
| 48 | |
| 49 | def __init__(self, model_path: str = None): |
| 50 | super().__init__() |
| 51 | self.model_path = model_path or self._get_default_path() |
| 52 | self._session = None |
| 53 | self._input_name = None |
| 54 | self._output_name = None |
| 55 | |
| 56 | def _get_default_path(self) -> str: |
| 57 | """Default model path under models/rmbg/.""" |
no test coverage detected