caffe下FCN注意事项

来源:互联网 发布:勒夏特列原理 知乎 编辑:程序博客网 时间:2024/05/16 14:20

FCN的分类的label是直接写类号码:1,2,3,...,255。

但softmaxwithLoss的第一个bottom,也就是网络输出,是要求向量。

即label=3的话,输出是0,0,0,1,0,0,...


所以写脚本注意一下,输出层的输出个数不能写1。

就是说softmaxwithLoss的两个bottom第二维数值是不一样的。

原创粉丝点击