- 博客(1)
- 资源 (2)
- 收藏
- 关注
原创 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
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人