Labelme+SAM_labelme sam

网上关于Labelme+SAM怎么安装的文章太旧了，不太适合，写个极简的，看不懂建议手动打标签

1、安装labelme，这个不用教了吧

2、卸载onnxruntime，换成onnxruntime-gpu版本

3、根据系统版本，配置cuda和cuDnn，看清楚版本选！选错了gpu用不了

4、添加onnx模型，把模型下载到C:\Users\你的用户.cache\gdown

SAM
https://www.aliyundrive.com/s/CQwyc9dLr3y
点击链接保存，或者复制本段内容，打开「阿里云盘」APP ，无需下载极速在线查看，视频原画倍速播放。
自己把后缀.exe删掉，这是onnx格式的文件

5、修改labelme文件
找到labelme的库文件，.\Anaconda3\envs\你的环境\Lib\site-packages\labelme\ai\models，修改sam.py(名称没打全，自己脑补)

class SegmentAnythingModel:
    def __init__(self, name, encoder_path, decoder_path):
        self.name = name

        self._image_size = 1024

        self._encoder_session = onnxruntime.InferenceSession(encoder_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
        self._decoder_session = onnxruntime.InferenceSession(decoder_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])

        self._lock = threading.Lock()
        self._image_embedding_cache = collections.OrderedDict()

        self._thread = None
1
2
3
4
5
6
7
8
9
10
11
12
13

自己比较多了那些东西

6、lableme里面可以用了