nebgnahz / darknet-rs Goto Github PK

fn cv_mat_to_darknet_image(mat: &cv::Mat) -> darknet::InputImage {
    let data: *const u8 = mat.data();
    let h = mat.rows;
    let w = mat.cols;
    let c = mat.channels;

    let mut out = darknet::InputImage::new(w, h, c);
    let out_data = out.data_mut();
    let mut count = 0;
    for k in 0..c {
        for y in 0..h {
            for x in 0..w {
                let offset = (c * (w * y + x) + k) as isize;
                unsafe {
                    let v = *(data.offset(offset)) as f32 / 255.0;
                    *out_data.offset(count) = v;
                }
                count += 1;
            }
        }
    }
    out
}