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

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。