Method __init__

flowmap/model/model.py:42–52

(
        self,
        cfg: ModelCfg,
        num_frames: int | None = None,
        image_shape: tuple[int, int] | None = None,
    )

class Model(nn.Module):
    def __init__(
        self,
        cfg: ModelCfg,
        num_frames: int | None = None,
        image_shape: tuple[int, int] | None = None,
    ) -> None:
        super().__init__()
        self.cfg = cfg
        self.backbone = get_backbone(cfg.backbone, num_frames, image_shape)
        self.intrinsics = get_intrinsics(cfg.intrinsics)
        self.extrinsics = get_extrinsics(cfg.extrinsics, num_frames)

    def forward(
        self,