def _filter()

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

• 14 lines of code
• 10 McCabe index (conditional complexity)

    def _filter(self, entity: AnnoEntity, ignored_gt_entities=None):
        if ignored_gt_entities is None:
            if self.is_train:
                # any entity whose visibility is below the pre-defined
                # threshold should be filtered out
                # meanwhile, any entity whose class does not have label
                # needs to be filtered
                if entity.blob['visibility'] < self.visibility_thresh or \
                        not any(k in ('person', '2', '7') for k in entity.labels):
                    return True
            else:
                if 'person' not in entity.labels or int(entity.id) < 0:
                    return True
        else:
            for entity_ in ignored_gt_entities:
                if bbs_iou(entity, entity_) >= self.iou_thresh:
                    return True
            return False