caffe2
gaomingda
这个作者很懒,什么都没留下…
展开
-
caffe2 Softmax, SoftmaxWithLoss 使用
本文主要记录caffe2的Softmax与SoftmaxWithLoss op 如何使用(输入格式), 特别介绍如何应用在FCN的语义分割中. 1 Softmax 输入:2D, 但可以不是,计算时代码会强制转换 输出: 与输入相同的维度 arg:axis 默认等于1, 其实没太搞懂 在FCN的输出probs时以下操作, 直接上测试的代码 from caffe2.python imp...原创 2018-05-31 20:21:34 · 1026 阅读 · 0 评论 -
caffe2 创建自己的Python层
python 层 一、 没有反向传播 例如: def mul_ab(inputs, outputs): num_blobs = int(len(inputs)/2) for i in range(num_blobs): # range(x), x must be integer i = int(i) # inputs 的坐标必须是整形 o...原创 2018-06-29 14:42:08 · 378 阅读 · 1 评论