自定义 Python 图层
- 使用 Python 层构建 Caffe
Caffe 需要使用 WITH_PYTHON_LAYER
选项进行编译:
WITH_PYTHON_LAYER=1 make && make pycaffe
- 我应该在哪里保存类文件?
你有两个选择(至少我知道)。你可以将自定义图层文件保存在与运行 caffe 命令相同的文件夹中(可能是你的原型文件文件所在的位置)。另一种方式,也是我最喜欢的方法,是将所有自定义图层保存在文件夹中,并将此文件夹添加到 PYTHONPATH 中。