伯克利caffe命令行界面

Bekeley caffe command line interface

如果您使用的是自定义 python 层 - 假设您在 python 中正确编写了 class - 假设 class 的名称是 "my_ugly_custom_layer";然后你在linux命令行界面执行caffe,

你如何确保 caffe 知道如何找到你为层编写 class 的文件?您只是将 .py 文件放在与 train.prototxt 相同的目录中吗?

如果您在 python 中编写了自定义 class,您需要使用 python 包装器接口吗?

您的 python 图层在 prototxt 中有两个参数:layer: 您定义实现图层的 python class 名称,moduule: 其中您定义实现层 class 的 .py 文件名。
当你 运行 caffe (从命令行或通过 python 界面)你需要确保你的模块在 PYTHONPATH