Python make_box_predictor示例

示例#1

文件： box_head.py 项目： shenmayufei/294-82

 def __init__(self, cfg):
     super().__init__()
     self.cfg = cfg
     self.predictor = make_box_predictor(cfg)
     self.loss_evaluator = MultiBoxLoss(neg_pos_ratio=cfg.MODEL.NEG_POS_RATIO)
     self.post_processor = PostProcessor(cfg)
     self.priors = None

示例#2

文件： box_head.py 项目： BarryKCL/OEDet_COCO512

 def __init__(self, cfg):
     super().__init__()
     self.cfg = cfg
     self.predictor = make_box_predictor(cfg)
     self.loss_evaluator = MultiBoxLoss(neg_pos_ratio=cfg.MODEL.NEG_POS_RATIO)
     self.mask_criterion = Mask_BCELoss(2, 0.5, True, 0, True, 3, 0.5, False)
     self.post_processor = PostProcessor(cfg)
     self.priors = None

示例#3

文件： box_head.py 项目： touchylk/lwk_SSD

 def __init__(self, cfg):
     super().__init__()
     self.cfg = cfg
     self.predictor = make_box_predictor(cfg)
     self.loss_evaluator = MultiBoxLoss(neg_pos_ratio=cfg.MODEL.NEG_POS_RATIO,num_classes=cfg.MODEL.NUM_CLASSES,
                                        alpha=None,gamma=2
                                        )
     self.post_processor = PostProcessor(cfg)
     self.priors = None

示例#4

    def __init__(self, cfg):
        super().__init__()
        self.cfg = cfg
        self.predictor = make_box_predictor(cfg)
        #
        if self.cfg.MODEL.BOX_HEAD.LOSS == 'FocalLoss':
            self.loss_evaluator = FocalLoss(0.25, 2)
        else:  # By default, we use MultiBoxLoss
            self.loss_evaluator = MultiBoxLoss(
                neg_pos_ratio=cfg.MODEL.NEG_POS_RATIO)

        self.post_processor = PostProcessor(cfg)
        self.priors = None