转换Open Model Zoo中fastseg-large模型时出现两类报错求解决方案
FastSeg-large模型转换报错解决方案
问题1:fastseg_large模块不存在报错
原因
Open Model Zoo的converter工具默认按照预设的模块名检索模型定义文件,你下载的模型定义文件存放在/model/fastseg/model子目录下,和工具预期的检索路径不匹配。
解决方法
二选一即可:
- 调整目录结构:进入
C:\Users\david\Desktop\Neural Zoo\open_model_zoo\models\public\fastseg-large\model目录,把fastseg文件夹内的所有模型定义文件直接移动到model根目录下 - 修改模型配置:找到fastseg-large对应的
model.yml配置文件,修改其中的模块导入路径,指向fastseg.model下的对应模型类即可
问题2:无法从torch._six导入container_abcs报错
原因
该报错是PyTorch版本不兼容导致,container_abcs模块在PyTorch 1.8及以上版本中已经从torch._six包中移除,迁移到了collections.abc下。
解决方法
二选一即可:
- 降级PyTorch版本:执行命令
pip install torch==1.7.1 torchvision==0.8.2,安装适配旧版导入规则的PyTorch版本 - 兼容高版本PyTorch:找到模型定义中包含
from torch._six import container_abcs的语句,替换为import collections.abc as container_abcs即可
内容的提问来源于stack exchange,提问作者user17535004
相关产品推荐
相关产品推荐

