自定义 Python 图层

- 使用 Python 层构建 Caffe

Caffe 需要使用 WITH_PYTHON_LAYER 选项进行编译:

WITH_PYTHON_LAYER=1 make && make pycaffe

- 我应该在哪里保存类文件?

你有两个选择(至少我知道)。你可以将自定义图层文件保存在与运行 caffe 命令相同的文件夹中(可能是你的原型文件文件所在的位置)。另一种方式,也是我最喜欢的方法,是将所有自定义图层保存在文件夹中,并将此文件夹添加到 PYTHONPATH 中。

参考

  1. 克里斯托弗布雷斯的博客
  2. Caffe Github
  3. 堆栈溢出