Skip to content

Commit

Permalink
add eval curve to metrics.json
Browse files Browse the repository at this point in the history
Reviewed By: rbgirshick

Differential Revision: D18033952

fbshipit-source-id: b835578d53372980ba9f1918b1229224774cebd2
  • Loading branch information
ppwwyyxx authored and facebook-github-bot committed Oct 21, 2019
1 parent 8863132 commit 462c903
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 6 deletions.
3 changes: 2 additions & 1 deletion MODEL_ZOO.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,8 @@ Unless otherwise noted, the following settings are used for all runs:
Although 1x models are heavily under-trained, we provide some ResNet-50 models with the 1x (~12 COCO epochs)
training schedule for comparison when doing quick research iteration.
* The *model id* column is provided for ease of reference.
To check downloaded file integrity: any model on this page contains its md5 prefix in its file name.
To check downloaded file integrity, any model on this page contains its md5 prefix in its file name.
Each model also comes with a metrics file with all the training statistics and evaluation curves.

#### ImageNet Pretrained Models

Expand Down
5 changes: 0 additions & 5 deletions detectron2/evaluation/panoptic_evaluation.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ def __init__(self, dataset_name, output_dir):
}

self._predictions_json = os.path.join(output_dir, "predictions.json")
self._predictions_dir = os.path.join(output_dir, "predictions")

def reset(self):
self._predictions = []
Expand Down Expand Up @@ -95,10 +94,6 @@ def evaluate(self):
gt_folder = self._metadata.panoptic_root

with tempfile.TemporaryDirectory(prefix="panoptic_eval") as pred_dir:
if "://" not in self._predictions_dir:
pred_dir = self._predictions_dir
os.makedirs(pred_dir, exist_ok=True)

logger.info("Writing all panoptic predictions to {} ...".format(pred_dir))
for p in self._predictions:
with open(os.path.join(pred_dir, p["file_name"]), "wb") as f:
Expand Down

0 comments on commit 462c903

Please sign in to comment.