Create evaluator for a COCO dataset. Currently only Mask R-CNN is supported, dataset of interest is COCO, so only COCOEvaluator is implemented.
(dataset_name)
| 35 | sys.exit(1) |
| 36 | |
| 37 | def build_evaluator(dataset_name): |
| 38 | """ |
| 39 | Create evaluator for a COCO dataset. |
| 40 | Currently only Mask R-CNN is supported, dataset of interest is COCO, so only COCOEvaluator is implemented. |
| 41 | """ |
| 42 | evaluator_type = MetadataCatalog.get(dataset_name).evaluator_type |
| 43 | if evaluator_type in ["coco"]: |
| 44 | return COCOEvaluator(dataset_name) |
| 45 | else: |
| 46 | raise NotImplementedError("Evaluator type is not supported") |
| 47 | |
| 48 | def setup(config_file, weights): |
| 49 | """ |