首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >ModuleNotFoundError: Python3.10和3.11中没有名为“segmentation_models_pytorch.unet”的模块

ModuleNotFoundError: Python3.10和3.11中没有名为“segmentation_models_pytorch.unet”的模块
EN

Data Science用户
提问于 2023-05-31 20:57:08
回答 2查看 21关注 0票数 0

我使用Python3.8环境和segmentation_models_pytorch aka smp训练了一个分段模型。当我保存它并在我的预测环境中加载(Python3.6和smp)时,它只在

代码语言:javascript
运行
复制
import torch

model = torch.load(path.join('models', model_name))

但是,它与onnx包(onnx需要更新的Python)发生冲突。我用Python3.10创建了新的conda环境(另一个使用Python3.11)。现在,torch拒绝装载错误消息ModuleNotFoundError: No module named 'segmentation_models_pytorch.unet'的模型。

正确的方法是什么?

  1. 安装支持torchtorch.load
  2. 了解unet架构的火炬
  3. 支持onnx模型转换的环境
  4. most需要从os导入路径、environ导入torch torch_model = torch.load(path.join('models‘、'my_model.pth'))追溯(最近调用的最后一个):文件"D:\workspace\acne_prod\pytorch2onnx.py",第3行, torch_model =torch_load(’模型‘,( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ "C:\Users\sixty\anaconda3\envs\acne_prod_smp_onnx\Lib\site-packages\torch\serialization.py","C:\Users\sixty\anaconda3\envs\acne_prod_smp_onnx\Lib\site-packages\torch\serialization.py",行809,加载返回_load(opened_zipfile,map_location,pickle_module,**pickle_load_args) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File“C:\Users\sixty\anaconda3\envs\acne_prod_smp_onnx\Lib\site-packages\torch\serialization.py”,第1172行,在"C:\Users\sixty\anaconda3\envs\acne_prod_smp_onnx\Lib\site-packages\torch\serialization.py",_load = unpickler.load() ^文件文件,在find_class返回的.find_class(mod_name,名称)中,没有名为'segmentation_models_pytorch.unet‘的模块。
EN

回答 2

Data Science用户

发布于 2023-06-01 06:03:54

Segmentation_models是一个不同于python的python包,需要单独安装。pip安装-U git+https://github.com/qubvel/segmentation_models.pytorch在python3.10.5及更高版本上为我工作。

票数 0
EN

Data Science用户

发布于 2023-06-01 19:10:18

这是对我有效的解决办法。

将您的SMP降级到您正在进行培训的版本。或者在资源允许的情况下,用新的SMP训练你的模型。这样您就可以避免遗留的Python问题。

因此。这不是火把问题。这是由于SMP模型的改变导致了与dackward兼容性的问题。

票数 0
EN
页面原文内容由Data Science提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://datascience.stackexchange.com/questions/121878

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档