UV-SAM
UV-SAM copied to clipboard
No luck to run inference_seg.py on colab
Tks for great job but so far can't run inference_seg.py on colab
/content/UV-SAM
/usr/local/lib/python3.10/dist-packages/mmcv/cnn/bricks/transformer.py:33: UserWarning: Fail to import ``MultiScaleDeformableAttention`` from ``mmcv.ops.multi_scale_deform_attn``, You should install ``mmcv`` rather than ``mmcv-lite`` if you need this module.
warnings.warn('Fail to import ``MultiScaleDeformableAttention`` from '
Traceback (most recent call last):
File "/content/UV-SAM/inference_seg.py", line 18, in <module>
from mmseg.apis import init_model, inference_model
File "/content/UV-SAM/./mmseg/apis/__init__.py", line 2, in <module>
from .inference import inference_model, init_model, show_result_pyplot
File "/content/UV-SAM/./mmseg/apis/inference.py", line 16, in <module>
from mmseg.models import BaseSegmentor
File "/content/UV-SAM/./mmseg/models/__init__.py", line 2, in <module>
from .backbones import * # noqa: F401,F403
File "/content/UV-SAM/./mmseg/models/backbones/__init__.py", line 7, in <module>
from .fast_scnn import FastSCNN
File "/content/UV-SAM/./mmseg/models/backbones/fast_scnn.py", line 7, in <module>
from mmseg.models.decode_heads.psp_head import PPM
File "/content/UV-SAM/./mmseg/models/decode_heads/__init__.py", line 2, in <module>
from .ann_head import ANNHead
File "/content/UV-SAM/./mmseg/models/decode_heads/ann_head.py", line 8, in <module>
from .decode_head import BaseDecodeHead
File "/content/UV-SAM/./mmseg/models/decode_heads/decode_head.py", line 14, in <module>
from ..losses import accuracy
File "/content/UV-SAM/./mmseg/models/losses/__init__.py", line 7, in <module>
from .focal_loss import FocalLoss
File "/content/UV-SAM/./mmseg/models/losses/focal_loss.py", line 6, in <module>
from mmcv.ops import sigmoid_focal_loss as _sigmoid_focal_loss
File "/usr/local/lib/python3.10/dist-packages/mmcv/ops/__init__.py", line 2, in <module>
from .active_rotated_filter import active_rotated_filter
File "/usr/local/lib/python3.10/dist-packages/mmcv/ops/active_rotated_filter.py", line 10, in <module>
ext_module = ext_loader.load_ext(
File "/usr/local/lib/python3.10/dist-packages/mmcv/utils/ext_loader.py", line 13, in load_ext
ext = importlib.import_module('mmcv.' + name)
File "/usr/lib/python3.10/importlib/__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ModuleNotFoundError: No module named 'mmcv._ext'
my requirements.txt
can be installed successfully on colab, python 310.
albumentations==1.3.1
cityscapesscripts==2.2.2
dadaptation==3.1
imagecorruptions==1.1.2
imgviz==1.7.2
instaboostfast==0.1.2
ipdb==0.13.13
lightning==2.0.1
lion_pytorch==0.1.2
matplotlib==3.7.2
memory_profiler==0.61.0
mmcls==0.25.0
mmcv==2.0.0
mmdet==3.3.0
mmengine==0.8.1
mmyolo==0.6.0
nibabel==5.1.0
numpy==1.24.1
onnxruntime==1.15.1
opencv_python==4.8.0.74
opencv_python_headless==4.8.0.74
packaging==23.1
Pillow==9.3.0
prettytable==3.8.0
psutil==5.9.5
pycocotools==2.0.6
rich==13.5.2
sahi==0.11.14
scikit_learn==1.3.0
scipy==1.11.2
segment_anything==1.0
setproctitle==1.3.2
Shapely==2.0.1
six==1.16.0
terminaltables==3.1.10
timm==0.9.5
torch==2.3.1
torchmetrics==0.11.4
torchvision
tqdm==4.65.0