Spaces:
Running
on
Zero
Running
on
Zero
File size: 732 Bytes
9e15541 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from .image_encoder import ImageEncoder
from .monodepth2 import Monodepth2
from .spatial_encoder import SpatialEncoder
from scenedino.models.backbones.dino.dinov2_module import DINOv2Module
def make_backbone(conf, **kwargs):
enc_type = conf.get("type", "monodepth2") # spatial | global
if enc_type == "monodepth2":
net = Monodepth2.from_conf(conf, **kwargs)
elif enc_type == "spatial":
net = SpatialEncoder.from_conf(conf, **kwargs)
elif enc_type == "global":
net = ImageEncoder.from_conf(conf, **kwargs)
elif enc_type == "dinov2":
net = DINOv2Module.from_conf(conf, **kwargs)
else:
raise NotImplementedError(f"Unsupported encoder type: {enc_type}")
return net
|