Class BackboneOutput

flowmap/model/backbone/backbone.py:15–17

@dataclass
class BackboneOutput:
    depths: Float[Tensor, "batch frame height width"]
    weights: Float[Tensor, "batch frame-1 height width"]


class Backbone(nn.Module, ABC, Generic[T]):