当前位置:   article > 正文

运行Grounded-Segment-Anything报错:NameError: name ‘_C‘ is not defined_grounded segment anything

grounded segment anything

没有使用docker部署,完全按照项目中的安装步骤进行安装的,运行项目中的grounded_sam_demo.py时出现如下报错:

  1. Traceback (most recent call last):
  2. File "/root/Downloads/Grounded-Segment-Anything/grounded_sam_demo.py", line 195, in <module>
  3. boxes_filt, pred_phrases = get_grounding_output(
  4. File "/root/Downloads/Grounded-Segment-Anything/grounded_sam_demo.py", line 66, in get_grounding_output
  5. outputs = model(image[None], captions=[caption])
  6. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
  7. return forward_call(*args, **kwargs)
  8. File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/groundingdino.py", line 313, in forward
  9. hs, reference, hs_enc, ref_enc, init_box_proposal = self.transformer(
  10. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
  11. return forward_call(*args, **kwargs)
  12. File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 258, in forward
  13. memory, memory_text = self.encoder(
  14. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
  15. return forward_call(*args, **kwargs)
  16. File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 576, in forward
  17. output = checkpoint.checkpoint(
  18. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/utils/checkpoint.py", line 249, in checkpoint
  19. return CheckpointFunction.apply(function, preserve, *args)
  20. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/autograd/function.py", line 506, in apply
  21. return super().apply(*args, **kwargs) # type: ignore[misc]
  22. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/utils/checkpoint.py", line 107, in forward
  23. outputs = run_function(*args)
  24. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
  25. return forward_call(*args, **kwargs)
  26. File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 785, in forward
  27. src2 = self.self_attn(
  28. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
  29. return forward_call(*args, **kwargs)
  30. File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py", line 338, in forward
  31. output = MultiScaleDeformableAttnFunction.apply(
  32. File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/autograd/function.py", line 506, in apply
  33. return super().apply(*args, **kwargs) # type: ignore[misc]
  34. File "/root/Downloads/Grounded-Segment-Anything/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py", line 53, in forward
  35. output = _C.ms_deform_attn_forward(
  36. NameError: name '_C' is not defined

最终按照下面的方法解决:

解决办法:

https://github.com/IDEA-Research/Grounded-Segment-Anything/issues/436icon-default.png?t=N7T8https://github.com/IDEA-Research/Grounded-Segment-Anything/issues/436问题出在GroundingDINO上

简单说就是先把GroundingDINO给卸载掉,然后cd到GroundingDINO的文件夹,执行

  1. python setup.py build
  2. python setup.py install

得到解决

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/814167
推荐阅读
相关标签