Method __init__

model/depth_wrapper.py:15–19

(self, model_name: str)

class DepthAnythingWrapper(nn.Module):
    def __init__(self, model_name: str):
        super().__init__()
        assert model_name in model_configs.keys(), f"model_name should be in {model_configs.keys()}"
        self.model = self._build_depth_anything(model_name)
        self._freeze()
    
    def _build_depth_anything(self, model_name):
        from importlib import import_module