(pred_config, output_file)
| 79 | |
| 80 | |
| 81 | def do_evaluate(pred_config, output_file): |
| 82 | num_tower = max(cfg.TRAIN.NUM_GPUS, 1) |
| 83 | graph_funcs = MultiTowerOfflinePredictor( |
| 84 | pred_config, list(range(num_tower))).get_predictors() |
| 85 | |
| 86 | for dataset in cfg.DATA.VAL: |
| 87 | logger.info("Evaluating {} ...".format(dataset)) |
| 88 | dataflows = [ |
| 89 | get_eval_dataflow(dataset, shard=k, num_shards=num_tower) |
| 90 | for k in range(num_tower)] |
| 91 | all_results = multithread_predict_dataflow(dataflows, graph_funcs) |
| 92 | output = output_file + '-' + dataset |
| 93 | DatasetRegistry.get(dataset).eval_inference_results(all_results, output) |
| 94 | |
| 95 | |
| 96 | def do_predict(pred_func, input_file): |
no test coverage detected