Ejemplo n.º 1
0
    def __init__(self, assigner_config):

        # some compositions
        self.similarity_calc = similarity_calc_builder.build(
            assigner_config['similarity_calc_config'])
        self.bbox_coder = bbox_coder_builder.build(
            assigner_config['coder_config'])
        self.matcher = matcher_builder.build(assigner_config['matcher_config'])
        self.analyzer = Analyzer()

        self.fg_thresh = assigner_config['fg_thresh']
        self.bg_thresh = assigner_config['bg_thresh']
        # self.clobber_positives = assigner_config['clobber_positives']
        self.iog_similarity_calc = similarity_calc_builder.build(
            {'type': 'iog'})
        self.iod_similarity_calc = similarity_calc_builder.build(
            {'type': 'iod'})
Ejemplo n.º 2
0
    def __init__(self, assigner_config):

        # some compositions
        self.similarity_calc = similarity_calc_builder.build(
            assigner_config['similarity_calc_config'])
        self.bbox_coder = bbox_coder_builder.build(
            assigner_config['coder_config'])
        self.matcher = matcher_builder.build(assigner_config['matcher_config'])
        self.analyzer = Analyzer()

        self.fg_thresh = assigner_config['fg_thresh']
        self.bg_thresh = assigner_config['bg_thresh']
Ejemplo n.º 3
0
    def __init__(self, assigner_config):

        # some compositions
        self.similarity_calc = similarity_calc_builder.build(
            assigner_config['similarity_calc_config'])
        self.bbox_coder = bbox_coder_builder.build(
            assigner_config['coder_config'])
        self.matcher = matcher_builder.build(assigner_config['matcher_config'])
        self.analyzer = Analyzer()

        self.fg_thresh = assigner_config['fg_thresh']
        self.bg_thresh = assigner_config['bg_thresh']
        if assigner_config.get('fake_match_thresh') is not None:
            self.fake_match_thresh = assigner_config['fake_match_thresh']
        else:
            self.fake_match_thresh = 0.7
Ejemplo n.º 4
0
    def __init__(self, assigner_config):

        # some compositions
        self.similarity_calc = similarity_calc_builder.build(
            assigner_config['similarity_calc_config'])
        self.bbox_coder = bbox_coder_builder.build(
            assigner_config['coder_config'])

        self.bbox_coder_3d = bbox_coder_builder.build({'type': 'bbox_3d'})
        self.keypoint_coder = bbox_coder_builder.build({
            'type': 'keypoint',
            'size': [56, 56]
        })
        self.matcher = matcher_builder.build(assigner_config['matcher_config'])
        self.analyzer = Analyzer()

        self.fg_thresh = assigner_config['fg_thresh']
        self.bg_thresh = assigner_config['bg_thresh']
Ejemplo n.º 5
0
    def __init__(self, assigner_config):

        # some compositions
        self.reg_similarity_calc = similarity_calc_builder.build(
            assigner_config['similarity_calc_config'])

        # make sure match boxes when they are enough close for classification
        self.cls_similarity_calc = CenterSimilarityCalc()
        # make sure more bbox can be optimized
        # self.reg_similarity_calc = ScaleSimilarityCalc()

        self.bbox_coder = bbox_coder_builder.build(
            assigner_config['coder_config'])
        self.matcher = matcher_builder.build(assigner_config['matcher_config'])
        self.analyzer = Analyzer()

        self.fg_thresh = assigner_config['fg_thresh']
        self.bg_thresh = assigner_config['bg_thresh']