Skip to content

namwooo/MetricsPerCategoryImplementation

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

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

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%