Confusion Matrix in Object Detection with TensorFlow
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
README.md
confusion_matrix.py

README.md

Confusion Matrix in Object Detection with TensorFlow

This script generates a confusion matrix using the detections record file generated by the /object_detection/inference/infer_detections.py script.

For a full description of the format of the matrix, and how it is computed, take a look at the Confusion Matrix in Object Detection with TensorFlow article.

To run this script you need a couple of things:

  • The label map used by your model — This is the proto-buff file that you created in order to train your model.

  • A detection record file — This is the file generated by using the /object_detection/inference/infer_detections.py. This script runs a TFRecord file through your model and saves the results in a detection record file.

Here is an example of running the script:

python confusion_matrix.py --detections_record=testing_detections.record --label_map=label_map.pbtxt

The script will print the confusion matrix along with precision and recall information to the standard output.