第一阶段rpn
训练的正负样本数据
- 正样本:
- 每一个gt box选择一个IOU最高的anchor
- 选择IOU大于0.7的也作为正样本
- 负样本:选择IOU小于0.3,其余可以忽略。
实际中负样本肯定远远超过正样本的数量,这个时候选择训练数量为256个
正样本控制在128个以内,正负样本总数量控制在256个。(比例大概控制在1:1)
第二阶段classifier阶段
训练的正负样本数据
- 正样本
- gt和roi的iou大于0.5 的为正
- 负样本
- gt和roi的iou小于0.5的为负样本
实际中负样本肯定远远超过正样本的数量,这个时候选择训练数量为256个
正样本控制在128个以内,正负样本总数量控制在256个。(比例大概控制在1:1)