赞
踩
没有使用docker部署,完全按照项目中的安装步骤进行安装的,运行项目中的grounded_sam_demo.py时出现如下报错:
- Traceback (most recent call last):
- File "/root/Downloads/Grounded-Segment-Anything/grounded_sam_demo.py", line 195, in <module>
- boxes_filt, pred_phrases = get_grounding_output(
- File "/root/Downloads/Grounded-Segment-Anything/grounded_sam_demo.py", line 66, in get_grounding_output
- outputs = model(image[None], captions=[caption])
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
- return forward_call(*args, **kwargs)
- File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/groundingdino.py", line 313, in forward
- hs, reference, hs_enc, ref_enc, init_box_proposal = self.transformer(
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
- return forward_call(*args, **kwargs)
- File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 258, in forward
- memory, memory_text = self.encoder(
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
- return forward_call(*args, **kwargs)
- File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 576, in forward
- output = checkpoint.checkpoint(
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/utils/checkpoint.py", line 249, in checkpoint
- return CheckpointFunction.apply(function, preserve, *args)
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/autograd/function.py", line 506, in apply
- return super().apply(*args, **kwargs) # type: ignore[misc]
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/utils/checkpoint.py", line 107, in forward
- outputs = run_function(*args)
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
- return forward_call(*args, **kwargs)
- File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 785, in forward
- src2 = self.self_attn(
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
- return forward_call(*args, **kwargs)
- File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py", line 338, in forward
- output = MultiScaleDeformableAttnFunction.apply(
- File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/autograd/function.py", line 506, in apply
- return super().apply(*args, **kwargs) # type: ignore[misc]
- File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py", line 53, in forward
- output = _C.ms_deform_attn_forward(
- NameError: name '_C' is not defined

最终按照下面的方法解决:
解决办法:
https://github.com/IDEA-Research/Grounded-Segment-Anything/issues/436https://github.com/IDEA-Research/Grounded-Segment-Anything/issues/436问题出在GroundingDINO上
简单说就是先把GroundingDINO给卸载掉,然后cd到GroundingDINO的文件夹,执行
- python setup.py build
- python setup.py install
得到解决
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。