Spaces:

HorizonRobotics
/

EmbodiedGen-Image-to-3D

Running on Zero

xinjie.wang commited on 2 days ago

Commit

e3ebdbb

1 Parent(s): 5638c1f

update

Files changed (1) hide show

embodied_gen/models/sam3d.py CHANGED Viewed

@@ -94,7 +94,8 @@ class Sam3dInference:
     ) -> dict:
         if isinstance(image, Image.Image):
             image = np.array(image)
-        image = self.merge_mask_to_rgba(image, mask)
         return self.pipeline.run(
             image,
             None,

     ) -> dict:
         if isinstance(image, Image.Image):
             image = np.array(image)
+        if mask is not None:
+            image = self.merge_mask_to_rgba(image, mask)
         return self.pipeline.run(
             image,
             None,