如果您使用自定义 python 层 - 并假设您在 python 中正确编写了该类 - 假设该类的名称为 “my_ugly_custom_layer”；然后在linux命令行界面中执行caffe，
如何确保 caffe 知道如何找到为图层编写类的文件？您只需将 .py 文件放在与 train.prototxt 相同的目录中吗？
或
如果您用 python 编写了自定义类，您需要使用 python 包装器接口(interface)吗？
最佳答案
您的 python 层在 prototxt 中有两个参数:layer:，您在其中定义实现层的 python 类名，以及 module:，您在其中定义 .py 文件实现图层类的名称。
当您运行caffe(从命令行或通过Python界面)时，您需要确保您的模块位于PYTHONPATH
关于python - 伯克利咖啡命令行界面，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/45942883/