facebookresearch / facebookresearch/segment-anything
There is a error when I run the command
- Dominant language
- Jupyter Notebook
- Stars
- 54.9k
- Forks
- 6.4k
- PR merge metrics
- No merged PRs in 30d
Description
I use RTX3090 test the command:
```
python scripts/amg.py --checkpoint "D:\segment\sam_vit_l_0b3195.pth" --model-type vit_l --input "D:\segment\photo-bridage.jpg" --output "D:\segment\out" ,
```
but I get this error:
```
"Traceback (most recent call last):
File "D:\segment\segment-anything\scripts\amg.py", line 238, in
main(args)
File "D:\segment\segment-anything\scripts\amg.py", line 221, in main
masks = generator.generate(image)
File "D:\ProgramData\miniconda3\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "D:\ProgramData\miniconda3\lib\site-packages\segment_anything\automatic_mask_generator.py", line 163, in generate mask_data = self._generate_masks(image)
File "D:\ProgramData\miniconda3\lib\site-packages\segment_anything\automatic_mask_generator.py", line 206, in _generate_masks
crop_data = self._process_crop(image, crop_box, layer_idx, orig_size)
File "D:\ProgramData\miniconda3\lib\site-packages\segment_anything\automatic_mask_generator.py", line 245, in _process_crop
batch_data = self._process_batch(points, cropped_im_size, crop_box, orig_size)
File "D:\ProgramData\miniconda3\lib\site-packages\segment_anything\automatic_mask_generator.py", line 297, in _process_batch
data.filter(keep_mask)
File "D:\ProgramData\miniconda3\lib\site-packages\segment_anything\utils\amg.py", line 49, in filter
self._stats[k] = v[torch.as_tensor(keep, device=v.device)]
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 12.61 GiB (GPU 0; 24.00 GiB total capacity; 18.41 GiB already allocated; 4.19 GiB free; 18.44 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF"
```
Contributor guide
Assessment
This issue has not been assessed yet.