(self, det_threshold=0.6, det_max_instances=None, bbox_order='dist_to_orig', coord_bins=1000, min_visibility=0.0, return_raw=False)
| 625 | class DetectionTransform(AbstractTransform): |
| 626 | |
| 627 | def __init__(self, det_threshold=0.6, det_max_instances=None, bbox_order='dist_to_orig', coord_bins=1000, min_visibility=0.0, return_raw=False): |
| 628 | self.det_threshold = det_threshold |
| 629 | self.det_max_instances = det_max_instances |
| 630 | self.coord_bins = coord_bins |
| 631 | self.min_visibility = min_visibility |
| 632 | self.return_raw = return_raw |
| 633 | |
| 634 | if bbox_order == 'area': |
| 635 | self.bbox_order = self.order_bboxes_by_area |
| 636 | elif bbox_order == 'score': |
| 637 | self.bbox_order = self.order_bboxes_by_score |
| 638 | elif bbox_order == 'random': |
| 639 | self.bbox_order = self.shuffle_bboxes |
| 640 | else: |
| 641 | self.bbox_order = self.order_bboxes_by_dist_to_orig |
| 642 | |
| 643 | @staticmethod |
| 644 | def order_bboxes_by_area(bboxes): |
nothing calls this directly
no outgoing calls
no test coverage detected