使用sam插件ISAT标注分割数据
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即可

效果展示

大功告成,觉得好的就点点赞,点点收藏,如有什么错误或者建议麻烦指出(字码错了也可以说,毕竟经常这样),谢谢
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐



所有评论(0)