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.