导入sahi.model模块报错：No module named 'sahi.model'求助

代码示例 #

import cv2
from sahi.predict import predict
from sahi.model import Yolov5DetectionModel
from sahi.utils.cv import read_image

错误信息 #

执行上述代码时，sahi.predict和sahi.utils可正常导入，但导入sahi.model时触发错误：
ModuleNotFoundError: No module named 'sahi.model'

已安装依赖 #

• pip install torch torchvision torchaudio
• pip install sahi
• pip install yolov5

版本信息 #

• Pip版本：23.1.2
• Python版本：3.11.4

已尝试的操作 #

将sahi降级至0.10.8版本（执行pip install sahi==0.10.8），但问题仍未解决。

解决方案 #

sahi的模块结构在版本迭代中做了调整，sahi.model模块已被移除，对应的模型类迁移到了sahi.models下的细分模块中。针对Yolov5模型，修改导入语句即可解决问题：

修改后的代码：

import cv2
from sahi.predict import predict
from sahi.models.yolov5 import Yolov5DetectionModel
from sahi.utils.cv import read_image

原理：在sahi的新版本（包括0.10.8）中，模型类按框架分类存放在sahi.models的子模块里，Yolov5对应的模型类路径为sahi.models.yolov5.Yolov5DetectionModel。

内容的提问来源于stack exchange，提问作者Berkhan Guzeller