分类层的 num_output 和 标签的值域 不符合:
a. 要知道imagenet是进行1000类的分类任务,我自己的数据是一个二分类,就一定要把最后‘fc8’InnerProduct的分类层的num_output: 2原来是1000,这个设成自己label的类别总数就可以。 b. 但是注意同时要修改train.prototxt和deploy.prototxt两个网络配置文件中的num_output
我按这个修改的,loss小了,之前也是87.3365
本文分享了在使用Caffe框架进行图像分类任务时,针对自定义数据集调整最后分类层num_output参数的经验。通过将预训练模型的分类数量从1000类调整为自定义数据集的2类,并同步更新训练与部署配置文件,成功降低了模型损失值。
分类层的 num_output 和 标签的值域 不符合:
a. 要知道imagenet是进行1000类的分类任务,我自己的数据是一个二分类,就一定要把最后‘fc8’InnerProduct的分类层的num_output: 2原来是1000,这个设成自己label的类别总数就可以。 b. 但是注意同时要修改train.prototxt和deploy.prototxt两个网络配置文件中的num_output
我按这个修改的,loss小了,之前也是87.3365

被折叠的 条评论
为什么被折叠?