openmmlab出现KeyError: ‘xxx is not in the model registry....‘

问题描述

在复现基于mmpose框架的算法时，运行程序出现KeyError: 'xxx is not in the model registry....'的问题，报错原因是自定义的backbone等结构或者某些当前代码使用的方法没有注册到现有的包中, 导致在import的时候无法导入该方法。

解决方案

• 找到requirements.txt或者相关安装依赖环境的txt文件,重复安装一边;
• 将该包中的所有方法重新安装即可.

cd mmpose
pip install -r requirements.txt
pip install -v -e . 

问题解决。