【问题标题】:DummyData layer equivalent in Caffe2?Caffe2 中的 DummyData 层等效?
【发布时间】:2017-06-15 12:28:14
【问题描述】:

出于性能原因,我正在尝试将相当复杂的卷积神经网络从 Caffe 转换为 Caffe2。我知道转换等效层的教程,但是,我在 Caffe 中的 DummyData 层使转换崩溃。虽然我知道我很可能不得不扩展教程脚本来解决这个问题,但我希望有人能指出我在 Caffe2 中构建等效层的正确方向。

我还必须转换几个不同的 python 层,所以我不可避免地需要修改 Caffe2 Github repo 中的教程;我只是不确定如何解决这个问题,并认为看看 DummyData 层是如何完成的会很有帮助。不幸的是,我无法访问我编写的代码,因为它在我的计算机上工作,但是,到目前为止,我所做的只是添加了一个新函数,其参数与赋予所有其他层的参数相似并创建一个新的运营商。我不确定这是否是正确的方法(因为我可能错过了一个大致相同的现有运营商)。

【问题讨论】:

    标签: python caffe pycaffe caffe2


    【解决方案1】:

    我没有对此进行测试,但它可能有效: 添加 UniformFillOp(生成虚拟数据),然后将其输出提供给 ImageInput(用于小批量和数据增强)。

    可用于生成虚拟数据的运算符列表:https://github.com/caffe2/caffe2/blob/master/caffe2/operators/filler_op.cc#L17-L25

    【讨论】:

      猜你喜欢
      • 2019-02-21
      • 2018-01-26
      • 1970-01-01
      • 2020-03-28
      • 2020-10-14
      • 1970-01-01
      • 2017-04-30
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多