(self, cfg: Config)
| 24 | |
| 25 | class ValidateModel(BaseModel): |
| 26 | def __init__(self, cfg: Config): |
| 27 | super().__init__(cfg) |
| 28 | self.cfg = cfg |
| 29 | if self.cfg.task.task == "validation": |
| 30 | self.validation_cfg = self.cfg.task |
| 31 | else: |
| 32 | self.validation_cfg = self.cfg.task.validation |
| 33 | self.metric = MeanAveragePrecision(iou_type="bbox", box_format="xyxy", backend="faster_coco_eval") |
| 34 | self.metric.warn_on_many_detections = False |
| 35 | self.val_loader = create_dataloader(self.validation_cfg.data, self.cfg.dataset, self.validation_cfg.task) |
| 36 | self.ema = self.model |
| 37 | |
| 38 | def setup(self, stage): |
| 39 | self.vec2box = create_converter( |
nothing calls this directly
no test coverage detected