def __init__(self, configer):
super(FasterRCNN, self).__init__()
self.configer = configer
self.backbone, self.classifier = VGGModel(configer)()
self.rpn = NaiveRPN(configer)
self.rpn_target_assigner = RPNTargetAssigner(configer)
self.roi_generator = FRROIGenerator(configer)
self.roi_sampler = FRROISampler(configer)
self.bbox_head = BBoxHead(configer, self.classifier)
self.det_loss = FasterRCNNLoss(self.configer)
def __init__(self, configer):
super(FasterRCNN, self).__init__()
self.configer = configer
self.backbone, self.classifier = VGGModel(configer)()
self.rpn = NaiveRPN(configer)
self.rpn_target_assigner = RPNTargetAssigner(configer)
self.roi_generator = FRROIGenerator(configer)
self.roi_sampler = FRROISampler(configer)
self.bbox_head = BBoxHead(configer, self.classifier)
self.valid_loss_dict = configer.get('loss', 'loss_weights', configer.get('loss.loss_type'))
def __init__(self, configer):
self.configer = configer
self.blob_helper = BlobHelper(configer)
self.det_visualizer = DetVisualizer(configer)
self.det_parser = DetParser(configer)
self.det_model_manager = ModelManager(configer)
self.test_loader = TestDataLoader(configer)
self.roi_sampler = FRROISampler(configer)
self.rpn_target_generator = RPNTargetAssigner(configer)
self.fr_priorbox_layer = FRPriorBoxLayer(configer)
self.fr_roi_generator = FRROIGenerator(configer)
self.device = torch.device('cpu' if self.configer.get('gpu') is None else 'cuda')
self.det_net = None
self._init_model()