Skip to content

Factory

dataphy.visionpack.adapters.factory

Classes

Functions

factory(bg_cfg)

Source code in src/dataphy/visionpack/adapters/factory.py
def factory(bg_cfg):
    name = bg_cfg.adapter
    if name == "greenaug":
        return GreenAugAdapter(**bg_cfg.greenaug)
    if name == "roboengine":
        return RoboEngineAdapter(**bg_cfg.roboengine)
    if name == "inpaint":
        return InpaintAdapter(**bg_cfg.inpaint)
    raise ValueError(f"Unknown background adapter: {name}")