Description:
huars 0.1.0
Hierarchical Uncertainty Aggregation
This repo implements hierarchical uncertainty aggregation from the paper Active Learning for Object Detection with Evidential Deep Learning and Hierarchical Uncertainty Aggregation by Park et al. (2022).
It is in written in Rust with bindings for Python.
Build
Run maturin build to build the wheel.
Usage
Import the package with import hua.
Create a hua.HUA object with iou_threshold and score_threshold.
Call hua.HUA.run() with:
bounding_boxes: Array of bounding boxes in the format [x1, y1, x2, y2] of shape (N, 4),
class_probabilities: Array of class probabilities of shape (N, C),
uncertainty_scores: Array of instance level uncertainty scores of shape (N,),
scales: Array of FPN scales from which the corresponding prediction was obtained of shape (N,).
The method returns the informativeness score of the image and the indices of the groups created by HUA.
Acknowledgements
Powerboxes: It was used to speed up the IOU calculation in the code.
License
For personal and professional use. You cannot resell or redistribute these repositories in their original state.
Customer Reviews
There are no reviews.
