Spaces:

HorizonRobotics
/

EmbodiedGen-Text-to-3D

Running on Zero

xinjie.wang commited on 1 day ago

Commit

2220050

1 Parent(s): b0e8715

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,