【问题标题】:Bekeley caffe command line interfaceBerkeley caffe 命令行界面
【发布时间】:2017-08-29 15:22:24
【问题描述】:

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

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

如果你在 python 中编写了一个自定义类,你需要使用 python 包装器接口吗?

【问题讨论】:

  • 非常感谢!我会尝试并发表评论。
  • 有关"Python"层的更多信息,请参见this answer

标签: python caffe layer


【解决方案1】:

您的 Python 层在 prototxt 中有两个参数:layer: 用于定义实现层的 Python 类名称,moduule: 用于定义实现层类的 .py 文件名。
当您运行caffe(从命令行或通过python接口)时,您需要确保您的模块位于PYTHONPATH

【讨论】:

    猜你喜欢
    • 2021-06-26
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2016-11-10
    • 2012-11-19
    • 1970-01-01
    相关资源
    最近更新 更多