def __init__(self, module):
super(RPNHeadWarper, self).__init__()
self.module = module
self.anchor_generator = build_warper(self.module.anchor_generator)
self.bbox_coder = build_warper(self.module.bbox_coder)
self.test_cfg = module.test_cfg
self.rpn_nms = BatchedNMS(0.0, self.test_cfg.nms_thr, -1)
def __init__(self, module, test_cfg):
super(BBoxHeadWarper, self).__init__()
self.module = module
self.bbox_coder = build_warper(self.module.bbox_coder)
self.test_cfg = test_cfg
self.num_classes = module.num_classes
self.rcnn_nms = BatchedNMS(test_cfg.score_thr, test_cfg.nms.iou_threshold, backgroundLabelId = module.num_classes)
def __init__(self, module):
super(AnchorFreeHeadWraper, self).__init__()
self.module = module
self.test_cfg = module.test_cfg
self.num_classes = self.module.num_classes
self.rcnn_nms = BatchedNMS(module.test_cfg.score_thr,
module.test_cfg.nms.iou_threshold,
backgroundLabelId=-1)
def __init__(self, module):
super(GARPNHeadWraper, self).__init__(module)
self.test_cfg = module.test_cfg
if 'nms' in self.test_cfg:
self.test_cfg.nms_thr = self.test_cfg.nms['iou_threshold']
if 'max_per_img' in self.test_cfg:
self.test_cfg.nms_post = self.test_cfg.max_per_img
self.test_cfg.max_num = self.test_cfg.max_per_img
self.rpn_nms = BatchedNMS(0.0, self.test_cfg.nms_thr, -1)
def __init__(self, module):
super(AnchorHeadWarper, self).__init__()
self.module = module
self.anchor_generator = build_warper(self.module.anchor_generator)
self.bbox_coder = build_warper(self.module.bbox_coder)
self.test_cfg = module.test_cfg
self.num_classes = self.module.num_classes
self.use_sigmoid_cls = self.module.use_sigmoid_cls
self.rcnn_nms = BatchedNMS(module.test_cfg.score_thr, module.test_cfg.nms.iou_threshold, backgroundLabelId = self.num_classes)
def __init__(self, module):
super(RPNHeadWraper, self).__init__()
self.module = module
self.anchor_generator = build_wraper(self.module.anchor_generator)
self.bbox_coder = build_wraper(self.module.bbox_coder)
self.test_cfg = module.test_cfg
if 'nms' in self.test_cfg:
self.test_cfg.nms_thr = self.test_cfg.nms['iou_threshold']
if 'max_per_img' in self.test_cfg:
self.test_cfg.nms_post = self.test_cfg.max_per_img
self.rpn_nms = BatchedNMS(0.0, self.test_cfg.nms_thr, -1)
def __init__(self, module):
super(YOLOV3HeadWraper, self).__init__()
self.module = module
self.anchor_generator = build_wraper(self.module.anchor_generator)
self.bbox_coder = build_wraper(self.module.bbox_coder)
self.featmap_strides = module.featmap_strides
self.num_attrib = module.num_attrib
self.num_levels = module.num_levels
self.test_cfg = module.test_cfg
self.num_classes = self.module.num_classes
self.rcnn_nms = BatchedNMS(module.test_cfg.score_thr, module.test_cfg.nms.iou_thr, backgroundLabelId = -1)
def __init__(self, module):
super(GuidedAnchorHeadWarper, self).__init__()
self.module = module
self.loc_filter_thr = module.loc_filter_thr
self.num_anchors = module.num_anchors
self.square_anchor_generator = build_warper(self.module.square_anchor_generator)
self.anchor_coder = build_warper(self.module.anchor_coder)
self.bbox_coder = build_warper(self.module.bbox_coder)
self.test_cfg = module.test_cfg
self.num_classes = self.module.num_classes
self.use_sigmoid_cls = self.module.use_sigmoid_cls
if "score_thr" in module.test_cfg and "nms" in module.test_cfg and "iou_threshold" in module.test_cfg.nms:
self.rcnn_nms = BatchedNMS(module.test_cfg.score_thr, module.test_cfg.nms.iou_threshold, backgroundLabelId = self.num_classes)
def __init__(self, module):
super(YOLOXHeadWraper, self).__init__()
self.module = module
self.prior_generator = build_wraper(self.module.prior_generator)
self.cls_out_channels = self.module.cls_out_channels
iou_thr = 0.7
if 'iou_thr' in module.test_cfg.nms:
iou_thr = module.test_cfg.nms.iou_thr
elif 'iou_threshold' in module.test_cfg.nms:
iou_thr = module.test_cfg.nms.iou_threshold
self.test_cfg = module.test_cfg
self.num_classes = self.module.num_classes
self.rcnn_nms = BatchedNMS(module.test_cfg.score_thr,
iou_thr,
backgroundLabelId=-1)
def __init__(self, module):
super(StandardRoIHeadWarper, self).__init__()
self.module = module
# self.bbox_roi_extractor = module.bbox_roi_extractor
self.bbox_roi_extractor = build_warper(module.bbox_roi_extractor)
self.bbox_head = module.bbox_head
if module.with_shared_head:
self.shared_head = module.shared_head
else:
self.shared_head = None
self.test_cfg = module.test_cfg
self.rcnn_nms = BatchedNMS(
module.test_cfg.score_thr,
module.test_cfg.nms.iou_threshold,
backgroundLabelId=self.bbox_head.num_classes)
def __init__(self, module):
super(CascadeRoIHeadWarper, self).__init__()
self.module = module
self.bbox_roi_extractor = [
build_warper(extractor) for extractor in module.bbox_roi_extractor
]
self.bbox_head = module.bbox_head
if module.with_shared_head:
self.shared_head = module.shared_head
else:
self.shared_head = None
self.test_cfg = module.test_cfg
self.num_stages = module.num_stages
self.rcnn_nms = BatchedNMS(
module.test_cfg.score_thr,
module.test_cfg.nms.iou_threshold,
backgroundLabelId=self.bbox_head[-1].num_classes)
def __init__(self, module):
super(GFLHeadWraper, self).__init__(module)
self.rcnn_nms = BatchedNMS(module.test_cfg.score_thr,
module.test_cfg.nms.iou_threshold,
backgroundLabelId=-1)
def __init__(self, module):
super(GARPNHeadWarper, self).__init__(module)
self.rpn_nms = BatchedNMS(0.0, self.test_cfg.nms_thr, -1)