目标跟踪:MTCNN报错Check failed: registry.count(type) == 0 (1 vs. 0)
错误描述
当我在终端下用caffe命令对带有PythonLayer的网络训练时,出现了如下错误。
该错误提示Layer的卷积类型已经注册过了。为什么会出现这个错误呢?
其实是,博主机子上有多个Caffe版本,该版本的pycaffe接口没有被指定正确。
解决方法
我的一种方法是在train.sh脚本文件开头,添加指定PYTHONPATH。
即,在train.sh文件中
#train.sh
export PYTHONPATH=/path/to/your/caffe/python/:$PYTHONPATH #把caffe/python路径加在PYTHONPATH最前面,优先检索
TOOLS=/path/to/your/caffe/build/tools #指定命令路径,即指定caffe版本
$TOOLS/caffe train --solver="" blablabla...
加完路径后一切就恢复正常啦。