(self, cfg)
| 278 | """ |
| 279 | |
| 280 | def __init__(self, cfg): |
| 281 | self.cfg = cfg.clone() # cfg can be modified by model |
| 282 | self.model = build_model(self.cfg) |
| 283 | self.model.eval() |
| 284 | if len(cfg.DATASETS.TEST): |
| 285 | self.metadata = MetadataCatalog.get(cfg.DATASETS.TEST[0]) |
| 286 | |
| 287 | checkpointer = DetectionCheckpointer(self.model) |
| 288 | checkpointer.load(cfg.MODEL.WEIGHTS) |
| 289 | |
| 290 | self.aug = T.ResizeShortestEdge( |
| 291 | [cfg.INPUT.MIN_SIZE_TEST, cfg.INPUT.MIN_SIZE_TEST], cfg.INPUT.MAX_SIZE_TEST |
| 292 | ) |
| 293 | |
| 294 | self.input_format = cfg.INPUT.FORMAT |
| 295 | assert self.input_format in ["RGB", "BGR"], self.input_format |
| 296 | |
| 297 | def __call__(self, original_image): |
| 298 | """ |
nothing calls this directly
no test coverage detected