(cfg, model)
| 107 | |
| 108 | |
| 109 | def do_test(cfg, model): |
| 110 | results = OrderedDict() |
| 111 | for dataset_name in cfg.DATASETS.TEST: |
| 112 | data_loader = build_detection_test_loader(cfg, dataset_name) |
| 113 | evaluator = get_evaluator( |
| 114 | cfg, dataset_name, os.path.join(cfg.OUTPUT_DIR, "inference", dataset_name) |
| 115 | ) |
| 116 | results_i = inference_on_dataset(model, data_loader, evaluator) |
| 117 | results[dataset_name] = results_i |
| 118 | if comm.is_main_process(): |
| 119 | logger.info("Evaluation results for {} in csv format:".format(dataset_name)) |
| 120 | print_csv_format(results_i) |
| 121 | if len(results) == 1: |
| 122 | results = list(results.values())[0] |
| 123 | return results |
| 124 | |
| 125 | |
| 126 | def do_train(cfg, model, resume=False): |
no test coverage detected