FasterRCNN----交替训练

交替训练Alternating training

第一步,我们训练RPN,该网络用ImageNet预训练的模型初始化,并端到端微调;

第二步,我们利用第一步的RPN生成的候选框,由Fast R-CNN训练一个单独的检测网络,这个检测网络同样是由ImageNet预训练的模型初始化的,这时候两个网络还没有共享卷积层;

第三步,我们用检测网络初始化RPN训练,但我们固定共享的卷积层(第二步得到的),并且只微调RPN独有的层,现在两个网络共享卷积层了;

第四步,保持共享的卷积层固定,微调Fast R-CNN的fc层。这样,两个网络共享相同的卷积层,构成一个统一的网络。

猜你喜欢

转载自www.cnblogs.com/mimandehuanxue/p/8996038.html
今日推荐