def _filter()

in siammot/data/adapters/handler/data_filtering.py [0:0]

• 17 lines of code
• 9 McCabe index (conditional complexity)

    def _filter(self, entity: AnnoEntity, ignored_gt_entities=None):
        if ignored_gt_entities is None:
            range_distance_m = np.inf
            if 'range_distance_m' in entity.blob:
                range_distance_m = entity.blob['range_distance_m']

            labels = []
            if entity.labels is not None:
                labels = entity.labels

            if ('intruder' not in labels) or \
                    (range_distance_m >= self.range_distance_thresh):
                return True
        else:
            for entity_ in ignored_gt_entities:
                if entity_.bbox is not None:
                    if bbs_iou(entity, entity_) >= self.iou_thresh:
                        return True
        return False