similari 0.20.4

Machine learning framework for building object trackers and similarity search engines
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use crate::utils::bbox::Universal2DBox;
use crate::utils::nms::{nms, parallel_nms};
use pyo3::prelude::*;

#[pyfunction]
#[pyo3(name = "nms")]
pub fn nms_py(
    detections: Vec<(Universal2DBox, Option<f32>)>,
    nms_threshold: f32,
    score_threshold: Option<f32>,
) -> Vec<Universal2DBox> {
    nms(&detections, nms_threshold, score_threshold)
        .into_iter()
        .cloned()
        .collect()
}

#[pyfunction]
#[pyo3(name = "parallel_nms")]
pub fn parallel_nms_py(
    detections: Vec<(Universal2DBox, Option<f32>)>,
    nms_threshold: f32,
    score_threshold: Option<f32>,
) -> Vec<Universal2DBox> {
    parallel_nms(&detections, nms_threshold, score_threshold)
        .into_iter()
        .cloned()
        .collect()
}