目录
sam部署教程:
Ascend/ModelZoo-PyTorch - Gitee.com
segment-anyghing地址:
https://github.com/visheratin/segment-anything
https://github.com/OroChippw/segment-anything
ais_bench
sam onnx地址:
https://github.com/OroChippw/SegmentAnything-OnnxRunner?tab=readme-ov-file
报错:
import auto_optimizer
解决方法:安装msit
msit: 统一推理工具链入口,提供客户一体化开发工具,支持一站式调试调优 - Gitee.com
encode继续报错:
Traceback (most recent call last):
File "/root/lbg/project/ACL_PyTorch/built-in/cv/SAM/encoder_onnx_modify.py", line 158, in <module>
graph = fa_adapter.adapt()
File "/root/lbg/project/ACL_PyTorch/built-in/cv/SAM/encoder_onnx_modify.py", line 29, in adapt
self.__adapt_layer(softmax)
File "/root/lbg/project/ACL_PyTorch/built-in/cv/SAM/encoder_onnx_modify.py", line 72, in __adapt_layer
target_shape = self.__infer_target_shape(layer_name_prefix, q)
File "/root/lbg/project/ACL_PyTorch/built-in/cv/SAM/encoder_onnx_modify.py", line 110, in __infer_target_shape
shape_q = self.__graph.add_node(
File "/usr/local/lib/python3.10/dist-packages/auto_optimizer/graph_refactor/onnx/graph.py", line 402, in add_node
return self._add_node(node)
File "/usr/local/lib/python3.10/dist-packages/auto_optimizer/graph_refactor/interface/base_graph.py", line 569, in _add_node
raise ValueError("node name '{}' already exists!".format(node.name))
ValueError: node name 'Shape_q' already exists!
segment_anything_diff.patch
ACL_PyTorch/built-in/cv/SAM/segment_anything_diff.patch
cv报错:
Traceback (most recent call last):
File "/root/lbg/sam_quant_preprocessing.py", line 19, in <module>
import cv2
File "/usr/local/lib/python3.10/dist-packages/cv2/__init__.py", line 181, in <module>
bootstrap()
File "/usr/local/lib/python3.10/dist-packages/cv2/__init__.py", line 153, in bootstrap
native_module = importlib.import_module("cv2")
File "/usr/lib/python3.10/importlib/__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: libGL.so.1: cannot open shared object file: No such file or directory
sudo apt-get update
sudo apt-get install libgl1-mesa-glx