Pytorch ModuleNotFoundError: No module named ‘models‘

问题

今天，在使用pytorch深度学习框架训练出来的模型文件，在另外的工程中使用，碰到如下的错误：

(shennong) ubuntu@ubuntu:~/zy/medicinebox_detector$ python interface_demo.py
Using torch 1.7.1 CUDA:0 (GeForce RTX 2080 Ti, 11016MB)

Traceback (most recent call last):
File “interface_demo.py”, line 13, in
detector = MedicineBoxDetector(save_img=False, weights=’/home/ubuntu/ModelWeight/yolov5x1280fillPolyBaseManual.pt’)
File “/home/ubuntu/zy/medicinebox_detector/medicine_detection/medicinebox_detector.py”, line 35, in init
self.model = attempt_load(weights, map_location=self.device) # load FP32 model
File “/home/ubuntu/zy/medicinebox_detector/medicine_detection/models/experimental.py”, line 118, in attempt_load
model.append(torch.load(w, map_location=map_location)[‘model’].float().fuse().eval()) # load FP32 model
File “/home/ubuntu/.conda/envs/shennong/lib/python3.6/site-packages/torch/serialization.py”, line 594, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File “/home/ubuntu/.conda/envs/shennong/lib/python3.6/site-packages/torch/serialization.py”, line 853, in _load
result = unpickler.load()
ModuleNotFoundError: No module named ‘models’

下面是我的models所在的目录

├── interface_demo.py
…
├── medicine_detection
…
│ ├── models
│ │ ├── common.py
│ │ ├── experimental.py
│ │ ├── export.py
…

解决方法

这个问题在https://github.com/ultralytics/yolov5/issues/353这个地方已经解决
我参考

import sys
sys.path.insert(0, './yolov5')

我把interface_demo.py这个文件加如下行，问题解决：

import sys
sys.path.insert(0, '/home/ubuntu/zy/medicinebox_detector/medicine_detection')

Pytorch ModuleNotFoundError: No module named ‘models‘相关推荐

关于result = unpickler.load()ModuleNotFoundError: No module named ‘models‘解决
如果保存模型时使用 torch.save(model, path) 在加载时使用 model = torch.load(path) 可能会出现如下错误,通常这种错误就是把一个项目保存的权重文件用在另一 ...
PyTorch中的No module named models?
PyTorch中的No module named models? 原创迷途小书童的Note 迷途小书童的Note 2020/07/28 16:21 阅读数 1.4K 本文被收录于专区 AI & ...
完美解决**ModuleNotFoundError: No module named ‘src‘**
完美解决ModuleNotFoundError: No module named 'src' 运行src文件夹下的文件时,一直报错No module named 'src',我想的是src是一个文件夹 ...
pip install models ModuleNotFoundError: No module named ‘base‘
pip install models ModuleNotFoundError: No module named 'base' python -m pip install --upgrade --fo ...
pytorch中ModuleNotFoundError: No module named ‘tensorboard‘
from torch.utils.tensorboard import SummaryWriter 报错: ModuleNotFoundError: No module named 'tensorbo ...
ModuleNotFoundError: No module named 'torch'（anaconda安装好Pytorch却无法在Jupyter notebook使用）
Windows下使用anaconda安装配置CPU版的pytorch请参看这篇博客: https://blog.csdn.net/weixin_43876801/article/details/103 ...
安装pytorch时，在安装 future-0.18.2.tar.gz (829 kB)时报错：ModuleNotFoundError: No module named ‘_ctypes’解决办法
ModuleNotFoundError: No module named '_ctypes' from _ctypes import Union, Structure, Array ModuleNot ...
jupyter notebook运行出错：ModuleNotFoundError: No module named ‘keras‘ 解决办法
接上一篇 <win10+python3.6+tensorflow-cpu+keras+Pycharm环境下的tensorflow配置方法> 通过Anaconda Prompt 打开查看当 ...
支持向量机ModuleNotFoundError: No module named ‘sklearn.datasets.samples_generator‘
最近在用Anaconda中Pytorch学习SVM分类时,导入模块 from sklearn.datasets.samples_generator import make_blobs 时报错: Mod ...

Pytorch ModuleNotFoundError: No module named ‘models‘

Pytorch ModuleNotFoundError: No module named ‘models‘相关推荐

最新文章

热门文章