Skip to main content

detect_from_heatmap

Function detect_from_heatmap 

Source
pub fn detect_from_heatmap(
    heatmap: &Tensor,
    score_threshold: f32,
    min_area: usize,
    iou_threshold: f32,
    max_detections: usize,
) -> Result<Vec<Detection>, DetectError>
Expand description

Connected-component detector over heatmaps [H, W, 1].