[Fix] fix_SAM (#3149) (#3150)

Author: shiyutang

contrib/SegmentAnything/segment_anything/automatic_mask_generator.py

@@ -166,7 +166,6 @@ def generate(self, image: np.ndarray) -> List[Dict[str, Any]]:
                crop_box (list(float)): The crop of the image used to generate
                  the mask, given in XYWH format.
         """
-        paddle.device.set_device('gpu')
         # Generate masks
         mask_data = self._generate_masks(image)
 

contrib/SegmentAnything/segment_anything/modeling/image_encoder.py

@@ -437,9 +437,6 @@ def forward(self, x: paddle.Tensor) -> paddle.Tensor:
     import torch
     from segment_anything.modeling import ImageEncoderViT as ImageEncoderViT_torch
 
-    # Set random seed
-    # np.random.seed(42)
-    paddle.set_device('gpu')
     image_encoder_t = ImageEncoderViT_torch(
         depth=12,
         embed_dim=768,
@@ -454,7 +451,6 @@ def forward(self, x: paddle.Tensor) -> paddle.Tensor:
         global_attn_indexes=[2, 5, 8, 11],
         window_size=14,
         out_chans=256, )
-    # image_encoder_t = image_encoder_t.to('cuda:1')
 
     image_encoder = ImageEncoderViT(
         depth=12,

contrib/SegmentAnything/segment_anything/modeling/sam.py

@@ -63,7 +63,10 @@ def __init__(
 
     @property
     def device(self) -> Any:
-        return 'gpu'
+        if paddle.is_compiled_with_cuda():
+            return 'gpu'
+        else:
+            return 'cpu'
 
     @paddle.no_grad()
     def forward(

contrib/SegmentAnything/segment_anything/predictor.py

@@ -270,7 +270,10 @@ def get_image_embedding(self) -> paddle.Tensor:
 
     @property
     def device(self) -> paddle.device:
-        return 'gpu'
+        if paddle.is_compiled_with_cuda():
+            return 'cpu'
+        else:
+            return 'gpu'
 
     def reset_image(self) -> None:
         """Resets the currently set image."""