third_party/sam-3d-objects/sam3d_objects/pipeline/depth_models/moge.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
from .base import DepthModel

class MoGe(DepthModel):
    def __call__(self, image):
        output = self.model.infer(
            image.to(self.device), force_projection=False
        )
        pointmaps = output["points"]
        output["pointmaps"] = pointmaps
        return output