similari 0.21.2

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
30
31
32
33
34
35
36
37
38
39
40
41

use crate::utils::bbox::Universal2DBox;
use crate::utils::nms::nms;
use pyo3::prelude::*;

#[pyfunction]
#[pyo3(
    name = "nms",
    text_signature = "(detections, nms_threshold, score_threshold)"
)]
pub fn nms_py(
    detections: Vec<(Universal2DBox, Option<f32>)>,
    nms_threshold: f32,
    score_threshold: Option<f32>,
) -> Vec<Universal2DBox> {
    let gil = Python::acquire_gil();
    let py = gil.python();

    py.allow_threads(|| {
        nms(&detections, nms_threshold, score_threshold)
            .into_iter()
            .cloned()
            .collect()
    })
}

//
// #[pyfunction]
// #[pyo3(
//     name = "parallel_nms",
//     text_signature = "(detections, nms_threshold, score_threshold)"
// )]
// 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()
// }