1、ISAT介绍

ISAT(Image Segmentation Annotation Tool)结合了Meta推出的Segment Anything Model(SAM),是一款功能强大且灵活的图像分割和打标签工具。

2、前提所需

anaconda3、pycharm下载完成(最好不要最新版本,也不要太旧,本人使用2021.3)

3、ISAT模型、权重下载

1.模型下载:github下载地址

4、环境配置

4.1.下载ISAT所需虚拟环境

打开anaconda prompt

​​

cd进入你下载的ISAT文件夹中

创建ISAT环境:输入 conda create -n sam(代表环境名称)python=3.8(使用Python的版本),按y+回车

无脑复制

conda create -n sam python=3.8

​进入你刚安装的sam环境:conda activate sam(你的环境名称)

无脑复制

conda activate sam

进入后()会变成你环境名称 (判断是否安装成功)

4.2.下载CUDA(GPU所需)

cuda是连接GPU和模型训练的桥梁,pytorch是进入桥梁那段上坡的路(CPU训练慢,GPU快还好)(最好选择在虚拟环境安装,以下介绍为虚拟机安装过程)

查看cuda最高支持版本,win+R,输入cmd,打开命令窗口,输入nvidia-smi.exe,红框位置为最大可支持安装版本

无脑复制

conda install cudatoolkit=11.8

 安装后的结果

4.3.下载pytorch(GPU所需)

cuda是连接GPU和模型训练的桥梁,pytorch是进入桥梁那段上坡的路(pytorch库必须和cuda版本匹配)

无脑复制(CUDA 11.8)

 conda安装

conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia

 pip安装

pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118

安装后的结果(以yolov13举例)

或者去pytorch官网查找(只是教你怎么去找你所下载CUDA对应的pytorch,以防下载的别的版本的CUDA) pytorch官网下载方法

4.4.下载ISAT依赖库

首先到你下载的ISAT文件中找到requirements.txt文

​然后安装包里的库文件​

无脑复制

pip install -r requirements.txt

只要中间不报错即为下载成功

4.5. 检测是否安装正确

(环境已经配好添加好了,6搞完)

建立cuda.py

import torch
print('CUDA版本:',torch.version.cuda)
print('Pytorch版本:',torch.__version__)
print('显卡是否可用:','可用' if(torch.cuda.is_available()) else '不可用')
print('显卡数量:',torch.cuda.device_count())
print('当前显卡型号:',torch.cuda.get_device_name())
print('当前显卡的CUDA算力:',torch.cuda.get_device_capability())
print('当前显卡的总显存:',torch.cuda.get_device_properties(0).total_memory/1024/1024/1024,'GB')
print('是否支持TensorCore:','支持' if (torch.cuda.get_device_properties(0).major >= 7) else '不支持')
print('当前显卡的显存使用率:',torch.cuda.memory_allocated(0)/torch.cuda.get_device_properties(0).total_memory*100,'%')

5、pycharm添加解释器

点击文件、设置

​​​

找到python解释器,添加新的解释器

找到conda环境、使用现有环境,找到你前面配置的sam(环境名称)

 还有老版的pycharm,使用应该找Anaconda3\envs\yolov10(环境名称)\python.exe

(用yolov9举例)

​​

​​

点击应用​​​

6、运行ISAT

运行main.py即可

效果展示

大功告成,觉得好的就点点赞,点点收藏,如有什么错误或者建议麻烦指出(字码错了也可以说,毕竟经常这样),谢谢

​​​​​​

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐