LaotechLabs / LOGOS

A Brand Independent logo detection model
Apache License 2.0
28 stars 6 forks source link

ModuleNotFoundError: No module named 'sahi.model' #4

Open vishalbharadwaj2406 opened 1 year ago

vishalbharadwaj2406 commented 1 year ago

I tried running the 'inference.ipynb' file. I made sure the 2 model files were properly downloaded and were in this path: 'utils/models/logo_model'. I ran all the cells and finally ran this cell:

from utils.logodetect import logodetection

model = logodetection()

I ran into this error:

ModuleNotFoundError                       Traceback (most recent call last)
 in 
      1 ## Initialise the logodetection model, checkpoints already provided
----> 2 from utils.logodetect import logodetection
      3 
      4 model = logodetection()

[/mnt/HDD/Pronisi/LOGO/temp/LOGOS/utils/logodetect.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/LOGOS/utils/logodetect.py) in 
      1 from unittest import result
      2 import torch
----> 3 from icevision.all import *
      4 import pandas as pd
      5 from icevision.models.checkpoint import model_from_checkpoint

[~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/__init__.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/__init__.py) in 
      5 from icevision.data import *
      6 from icevision import backbones
----> 7 from icevision import models
      8 from icevision.metrics import *
      9 from icevision.visualize import *

[~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/__init__.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/__init__.py) in 
     30 
     31 if SoftDependencies.sahi:
---> 32     from icevision.models import inference_sahi

[~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/inference_sahi.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/inference_sahi.py) in 
     12 
     13 
---> 14 from sahi.model import DetectionModel
     15 from sahi.prediction import ObjectPrediction
     16 from sahi.predict import get_sliced_prediction as sahi_get_sliced_prediction

ModuleNotFoundError: No module named 'sahi.model'

I verified the ice vision version I have is the same one mentioned in the readme. Request you to help me out with this issue.

duonglong289 commented 1 year ago

I tried running the 'inference.ipynb' file. I made sure the 2 model files were properly downloaded and were in this path: 'utils/models/logo_model'. I ran all the cells and finally ran this cell:

from utils.logodetect import logodetection

model = logodetection()

I ran into this error:

ModuleNotFoundError                       Traceback (most recent call last)
 in 
      1 ## Initialise the logodetection model, checkpoints already provided
----> 2 from utils.logodetect import logodetection
      3 
      4 model = logodetection()

[/mnt/HDD/Pronisi/LOGO/temp/LOGOS/utils/logodetect.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/LOGOS/utils/logodetect.py) in 
      1 from unittest import result
      2 import torch
----> 3 from icevision.all import *
      4 import pandas as pd
      5 from icevision.models.checkpoint import model_from_checkpoint

[~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/__init__.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/__init__.py) in 
      5 from icevision.data import *
      6 from icevision import backbones
----> 7 from icevision import models
      8 from icevision.metrics import *
      9 from icevision.visualize import *

[~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/__init__.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/__init__.py) in 
     30 
     31 if SoftDependencies.sahi:
---> 32     from icevision.models import inference_sahi

[~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/inference_sahi.py](https://file+.vscode-resource.vscode-cdn.net/mnt/HDD/Pronisi/LOGO/temp/~/miniconda3/envs/logo/lib/python3.7/site-packages/icevision/models/inference_sahi.py) in 
     12 
     13 
---> 14 from sahi.model import DetectionModel
     15 from sahi.prediction import ObjectPrediction
     16 from sahi.predict import get_sliced_prediction as sahi_get_sliced_prediction

ModuleNotFoundError: No module named 'sahi.model'

I verified the ice vision version I have is the same one mentioned in the readme. Request you to help me out with this issue.

Using older version of sahi could resolve your problem.

pip install sahi==0.9.4