分类层的 num_output 和 标签的值域 不符合:
a. 要知道imagenet是进行1000类的分类任务,我自己的数据是一个二分类,就一定要把最后‘fc8’InnerProduct的分类层的num_output: 2原来是1000,这个设成自己label的类别总数就可以。 b. 但是注意同时要修改train.prototxt和deploy.prototxt两个网络配置文件中的num_output
我按这个修改的,loss小了,之前也是87.3365
分类层的 num_output 和 标签的值域 不符合:
a. 要知道imagenet是进行1000类的分类任务,我自己的数据是一个二分类,就一定要把最后‘fc8’InnerProduct的分类层的num_output: 2原来是1000,这个设成自己label的类别总数就可以。 b. 但是注意同时要修改train.prototxt和deploy.prototxt两个网络配置文件中的num_output
我按这个修改的,loss小了,之前也是87.3365