vietanhdev / anylabeling

Effortless AI-assisted data labeling with AI support from YOLO, Segment Anything, MobileSAM!!
https://anylabeling.nrl.ai
GNU General Public License v3.0
2k stars 218 forks source link

The program crashed when loading mobileSAM #160

Closed YangWulve closed 6 months ago

YangWulve commented 7 months ago

The progrgam crashed when loading mobileSAM for the first time. And will crash everytime after crashed. Windows10, AnyLabeling v0.3.3crashedMobileSAM

scottix commented 6 months ago

I was able to replicate.

  1. Initial start loaded fine.
  2. Secondary usage crashed.
Traceback (most recent call last):
  File "C:\Users\S\AppData\Local\Programs\Python\Python310\lib\site-packages\anylabeling\utils.py", line 15, in run
    self.func(*self.args, **self.kwargs)
  File "C:\Users\S\AppData\Local\Programs\Python\Python310\lib\site-packages\anylabeling\services\auto_labeling\model_manager.py", line 389, in _load_model
    if model_config["type"] == "yolov5":
TypeError: 'NoneType' object is not subscriptable