(self, cfg)
| 184 | """ |
| 185 | |
| 186 | def __init__(self, cfg): |
| 187 | self.cfg = cfg.clone() # cfg can be modified by model |
| 188 | self.model = build_model(self.cfg) |
| 189 | self.model.eval() |
| 190 | if len(cfg.DATASETS.TEST): |
| 191 | self.metadata = MetadataCatalog.get(cfg.DATASETS.TEST[0]) |
| 192 | |
| 193 | checkpointer = DetectionCheckpointer(self.model) |
| 194 | checkpointer.load(cfg.MODEL.WEIGHTS) |
| 195 | |
| 196 | self.aug = T.ResizeShortestEdge( |
| 197 | [cfg.INPUT.MIN_SIZE_TEST, cfg.INPUT.MIN_SIZE_TEST], cfg.INPUT.MAX_SIZE_TEST |
| 198 | ) |
| 199 | |
| 200 | self.input_format = cfg.INPUT.FORMAT |
| 201 | assert self.input_format in ["RGB", "BGR"], self.input_format |
| 202 | |
| 203 | def __call__(self, original_image): |
| 204 | """ |
nothing calls this directly
no test coverage detected