在上篇CAFFE的基础上,我们继续喝咖啡。
ann4brains是一套基于CAFFE的代码,链接如下。
https://github.com/jeremykawahara/ann4brains
把仓库克隆下来之后,激活Python虚拟环境并cd到仓库里的example文件夹。
直接python helloworld.py
会提示你安装matplotlib和h5py。
直接pip install
即可。
继续python helloworld.py
会报错,如下所示。
为方便搜索,报错信息贴在下面。
TypeError: No loop matching the specified signature and casting was found for ufunc add
本质上是类型出错,在网上搜了搜相关信息,发现在Windows下变量类型会发生奇怪的改变,原因不明,故出此下策。
把虚拟环境下lib\site-packages\scipy\stats\stats.py
里的第3519行和第3520行修改为如下内容。
x.mean(dtype=float)
y.mean(dtype=float)
接着在第3540行
r = np.dot(xm/normxm, ym/normym)
之前增加如下内容。
xm = np.squeeze(np.asarray(xm))
ym = np.squeeze(np.asarray(ym))
即可顺利运行helloworld.py。
此方法有点野,可能会引起其他错误,不建议参考。