zychen2016 / MetricsPerCategoryImplementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implemetation of "include_metrics_per_category" in "eval_config" of tensorflow object detection api.Three code additions are listed below:
1.addtion in "tensorflow/models/research/pycocotools/cocoeval.py":from line 499 to line 648
2.addtion in "tensorflow/models/research/object_detetcion/metrics/coco_tools.py":from line 240 to line 244
3.add
{
  metrics_set: "coco_detection_metrics" 
  include_metrics_per_category: true 
}
to "eval_config" part in pipeline config file.

The code changes are only tested in my code to train a "fastrcnn_resnet_101" model. You should see per category metrics stats in tensorboard if they work. Any feedback would be appreciated.

About


Languages

Language:Python 100.0%