不能加载自动分割

yatengLG / ISAT_with_segment_anything

Labeling tool with SAM(segment anything model),supports SAM, SAM2, sam-hq, MobileSAM EdgeSAM etc.交互式半自动图像标注工具

https://www.yatenglg.cn/isat/

Other

1.19k stars 129 forks source link

Closed Tonyjoker1 closed 2 months ago

Tonyjoker1 commented 2 months ago

屏幕截图 2024-06-20 113530 作者大大，您好预标注的图片和标注文件，类别和模型都设置好以后，自动加载的图标经常是灰色的，然后整个程序就卡住不能用，重新加载也是一样。这要怎么解决呢

yatengLG commented 2 months ago

你是cpu吧。

卡住应该是在加载模型。cpu的话，建议用小点儿的模型，sam-hq-tiny或mobile-sam。

Tonyjoker1 commented 2 months ago

是环境配置里的那个要变成GPU呢

yatengLG commented 2 months ago

是环境配置里的那个要变成GPU呢

什么意思？

Tonyjoker1 commented 2 months ago

就是怎样用GPU去运行程序呢

yatengLG commented 2 months ago

就是怎样用GPU去运行程序呢

你安装gpu版本的pytorch就可以用gpu了。

Tonyjoker1 commented 2 months ago

好的，感谢