Spaces:

HorizonRobotics
/

EmbodiedGen-Texture-Gen

Running on Zero

EmbodiedGen-Texture-Gen / thirdparty /sam3d /sam3d_objects /pipeline /depth_models /base.py

xinjie.wang

update

ff883d2 9 months ago

307 Bytes

# Copyright (c) Meta Platforms, Inc. and affiliates.
import torch


class DepthModel:
    def __init__(self, model, device="cuda"):
        self.model = model
        self.device = torch.device(device)
        self.model.to(self.device)
        self.model.eval()

    def __call__(self, image):
        pass

	# Copyright (c) Meta Platforms, Inc. and affiliates.
	import torch


	class DepthModel:
	def __init__(self, model, device="cuda"):
	self.model = model
	self.device = torch.device(device)
	self.model.to(self.device)
	self.model.eval()

	def __call__(self, image):
	pass