Grounded-Segment-Anything icon indicating copy to clipboard operation
Grounded-Segment-Anything copied to clipboard

SAM predictor Error

Open lxianl455 opened this issue 1 year ago • 3 comments

When I try to get a mask, I got an error: Traceback (most recent call last): File "grounded_sam_demo.py", line 206, in masks, _, _ = predictor.predict_torch( File "/data0/lxl/environment/anaconda3_22_5_11/envs/try_segment/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context return func(*args, **kwargs) File "/data1/lxl/Grounded-Segment-Anything/segment_anything/segment_anything/predictor.py", line 229, in predict_torch low_res_masks, iou_predictions = self.model.mask_decoder( File "/data0/lxl/environment/anaconda3_22_5_11/envs/try_segment/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "/data1/lxl/Grounded-Segment-Anything/segment_anything/segment_anything/modeling/mask_decoder.py", line 94, in forward masks, iou_pred = self.predict_masks( File "/data1/lxl/Grounded-Segment-Anything/segment_anything/segment_anything/modeling/mask_decoder.py", line 147, in predict_masks masks = (hyper_in @ upscaled_embedding.view(b, c, h * w)).view(b, -1, h, w) RuntimeError: cannot reshape tensor of 0 elements into shape [0, -1, 256, 256] because the unspecified dimension size -1 can be any value and is ambiguous

lxianl455 avatar Apr 11 '23 13:04 lxianl455