object_detection_confusion_matrix

KO | EN

gitlite — search

#python #hacktoberfest #react #ai #typescript #go #docker #awesome #cli #awesome-list #vue #security #mcp #openai

object_detection_confusion_matrix

(★ 91)

Python class for calculating confusion matrix for object detection task

object_detection_confusion_matrix Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Confusion Matrix for Object Detection

[![DOI](https://zenodo.org/badge/249415928.svg)](https://zenodo.org/badge/latestdoi/249415928)

The ConfusionMatrix class can be used to generate confusion matrix for the object detection task.

## Usage

In the test code, you need to declare the ConfusionMatrix class with the appropriate parameters.

`conf_mat = ConfusionMatrix(num_classes = 3, CONF_THRESHOLD = 0.3, IOU_THRESHOLD = 0.5)`

The class has a function called, process_batch, you can use it update the confusion matrix.

An example usage can be:

`conf_mat.process_batch(preds, gt_boxes)` where preds are predictions made by model, [N, 6] x1, y1, x2, y2, confidence, class and gt_boxes are the ground truth labels, [M, 4] x1, y1, x2, y2, class.

## References

This repository uses a function from Pytorch repository, https://github.com/pytorch/vision/blob/ae228fef1ce176fa3e3949f5db7b6e87a1e33065/torchvision/ops/boxes.py#L138

and

The code is very similar to the repository below and main contribution of this repository is ConfusionMatrix class can be used with all deep learning frameworks.

https://github.com/svpino/tf_object_detection_cm

## Citations

If you would like to cite the concept, please use this doi: DOI: 10.5281/zenodo.3724203

## Contributions

Thanks to [Berkay Ugur Senocak](https://4turkuaz.github.io) for his contributions for maintenance issues on the repository.

object_detection_confusion_matrix

object_detection_confusion_matrix Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

object_detection_confusion_matrix

object_detection_confusion_matrix Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README