MaskRcnn-benchmark roi-box-head gt

box_head.py

line 39:

if self.training:

with torch.no_grad():

proposal = self.loss_evaluator.subsample(proposal,targets)---这里主要是通过rpn的正样本的anchors box来和targets 进行iou来再次将target的labels属性赋予box,背景是0,不要的是-1,其余的是81种的一个,这样就构成了在box-head中gt,再和roi的class,box进行loss

你可能感兴趣的:(MaskRcnn-benchmark roi-box-head gt)