Because infer_detections.py from the TFODAPI repo doesn't yet work for TF2, I haven't been able to use the original confusion matrix script. This script basically does the same thing as the original one, except it takes in the input tfrecord file and the model pb file so that the confusion matrix can be computed on detections directly instead of reading from a second record file. Until they update the TFODAPI this should suffice as a workaround.
Also, I added an option to visualize the detection boxes overlayed with the ground truth boxes and save the result. Good for sanity checks.
Because infer_detections.py from the TFODAPI repo doesn't yet work for TF2, I haven't been able to use the original confusion matrix script. This script basically does the same thing as the original one, except it takes in the input tfrecord file and the model pb file so that the confusion matrix can be computed on detections directly instead of reading from a second record file. Until they update the TFODAPI this should suffice as a workaround.
Also, I added an option to visualize the detection boxes overlayed with the ground truth boxes and save the result. Good for sanity checks.