44 lines
1.3 KiB
Rust
44 lines
1.3 KiB
Rust
use super::param::Treshold;
|
|
use opencv::core::{self, bitwise_and, in_range, Mat, Scalar, Size_};
|
|
use opencv::imgproc;
|
|
use opencv::Result;
|
|
|
|
pub fn image_warp(img: &Mat, homography: &Mat, h_size: Size_<i32>) -> Result<Mat> {
|
|
let mut warped_image = Mat::default();
|
|
imgproc::warp_perspective(
|
|
&img,
|
|
&mut warped_image,
|
|
homography,
|
|
h_size,
|
|
imgproc::INTER_CUBIC, // I dont see difference with INTER_CUBIC
|
|
core::BORDER_CONSTANT,
|
|
Scalar::default(),
|
|
)?;
|
|
|
|
Ok(warped_image)
|
|
}
|
|
|
|
pub fn image_treshold(img: &Mat, tresh: &Treshold) -> Result<Mat> {
|
|
let (t1, s1, l1) = (tresh.min_0 as f64, tresh.min_1 as f64, tresh.min_2 as f64);
|
|
let (t2, s2, l2) = (tresh.max_0 as f64, tresh.max_1 as f64, tresh.max_2 as f64);
|
|
let min = Mat::from_slice(&[t1, s1, l1])?;
|
|
let max = Mat::from_slice(&[t2, s2, l2])?;
|
|
let mut color_selected = Mat::default();
|
|
let _ = in_range(img, &min, &max, &mut color_selected);
|
|
let mut bord_treshed = Mat::default();
|
|
bitwise_and(&img, &img, &mut bord_treshed, &color_selected)?;
|
|
|
|
Ok(bord_treshed)
|
|
}
|
|
|
|
pub fn image_warp_treshold(
|
|
img: &Mat,
|
|
homography: &Mat,
|
|
h_size: Size_<i32>,
|
|
tresh: &Treshold,
|
|
) -> Result<Mat> {
|
|
let warped = image_warp(img, homography, h_size)?;
|
|
let treshed = image_treshold(&warped, tresh)?;
|
|
Ok(treshed)
|
|
}
|