Finalizing

.gitignore

@@ -4,4 +4,5 @@
 /converted_rgb.png
 .idea
 target
-app/target
+app/target
+dilated.jpg

Cargo.toml

@@ -18,4 +18,9 @@ exclude = ["*.jpg", "*.png"]
 [dependencies]
 colorutils-rs = "0.5.12"
 num-traits = "0.2.19"
-rayon = "1.10.0"
+rayon = "1.10.0"
+image = { version = "0.25.0", optional = true, default-features = false }
+
+[features]
+default = []
+image = ["dep:image"]

README.md

@@ -15,6 +15,25 @@ dilate_rgb(
 ).unwrap();
 ```
 
+#### Usage with image crate
+
+```rust
+let img = ImageReader::open("./assets/fruits.jpg")
+          .unwrap()
+          .decode()
+          .unwrap();
+let new_image = morphology_image(
+    img,
+    MorphOp::Dilate,
+    &structuring_element,
+    KernelShape::new(se_size, se_size),
+    BorderMode::default(),
+    MorphologyThreadingPolicy::default(),
+)
+.unwrap();
+new_image.save("dilated.jpg").unwrap();
+```
+
 ## Results
 
 Here is some examply bokeh effect
@@ -24,6 +43,13 @@ Here is some examply bokeh effect
     <img src="https://github.com/awxkee/fast_morphology/blob/master/assets/bokeh.jpg?raw=true" width="273" height="409">
 </p>
 
+And erosion
+
+<p float="left">
+    <img src="https://github.com/awxkee/fast_morphology/blob/master/assets/fruits.jpg?raw=true" width="273" height="409">
+    <img src="https://github.com/awxkee/fast_morphology/blob/master/assets/erosion.jpg?raw=true" width="273" height="409">
+</p>
+
 # Benchmarking
 
 If you wish to run benchmarks then

app/Cargo.toml

@@ -5,9 +5,9 @@ edition = "2021"
 
 [dependencies]
 image = "0.25.2"
-fast_morphology = {path = "../"}
+fast_morphology = {path = "../", features = ["image"]}
 imageproc = "0.25.0"
-opencv = {version = "0.93.0", features = ["imgproc"]}
+opencv = {version = "0.93.0", features = ["imgproc", "clang-runtime"]}
 
 [dev-dependencies]
 criterion = {version = "0.5.1", features = ["html_reports"]}

app/benches/dilation/main.rs

@@ -321,18 +321,18 @@ pub fn criterion_benchmark(c: &mut Criterion) {
     exec_bench_rgb(c, 10);
     exec_bench_rgb(c, 20);
     exec_bench_rgb(c, 30);
-    //
-    // exec_bench_rgba(c, 4);
-    // exec_bench_rgba(c, 7);
-    // exec_bench_rgba(c, 10);
-    // exec_bench_rgba(c, 20);
-    // exec_bench_rgba(c, 30);
 
-    // exec_bench_gray(c, 4);
-    // exec_bench_gray(c, 7);
-    // exec_bench_gray(c, 10);
-    // exec_bench_gray(c, 20);
-    // exec_bench_gray(c, 30);
+    exec_bench_rgba(c, 4);
+    exec_bench_rgba(c, 7);
+    exec_bench_rgba(c, 10);
+    exec_bench_rgba(c, 20);
+    exec_bench_rgba(c, 30);
+
+    exec_bench_gray(c, 4);
+    exec_bench_gray(c, 7);
+    exec_bench_gray(c, 10);
+    exec_bench_gray(c, 20);
+    exec_bench_gray(c, 30);
 }
 
 criterion_group!(benches, criterion_benchmark);

app/src/main.rs

@@ -1,5 +1,6 @@
 use fast_morphology::{
-    dilate, dilate_rgb, dilate_rgba, BorderMode, ImageSize, KernelShape, MorphologyThreadingPolicy,
+    dilate, dilate_rgb, dilate_rgba, erode, erode_rgba, morphology_image, BorderMode, ImageSize,
+    KernelShape, MorphExOp, MorphologyThreadingPolicy,
 };
 use image::{EncodableLayout, GenericImageView, ImageReader};
 use opencv::core::{
@@ -65,7 +66,7 @@ fn gaussian_kernel(size: usize, sigma: f32) -> Vec<Vec<f32>> {
 }
 
 fn main() {
-    let radius_size = 55;
+    let radius_size = 10;
     let mut structuring_element = circle_se(radius_size);
 
     opencv::core::set_use_opencl(false).expect("Failed to disable OpenCL");
@@ -114,7 +115,7 @@ fn main() {
 
     let image_size = ImageSize::new(dimensions.0 as usize, dimensions.1 as usize);
 
-    dilate(
+    erode(
         &channel_1_src,
         &mut channel_1_dst,
         image_size,
@@ -125,7 +126,7 @@ fn main() {
     )
     .unwrap();
 
-    dilate(
+    erode(
         &channel_2_src,
         &mut channel_2_dst,
         image_size,
@@ -136,7 +137,7 @@ fn main() {
     )
     .unwrap();
 
-    dilate(
+    erode(
         &channel_3_src,
         &mut channel_3_dst,
         image_size,
@@ -166,7 +167,7 @@ fn main() {
     let mut dst = vec![0u8; rgba_image.len()];
 
     let exec_time = Instant::now();
-    dilate_rgba(
+    erode_rgba(
         &rgba_image,
         &mut dst,
         image_size,
@@ -207,7 +208,7 @@ fn main() {
     let exec_time = Instant::now();
 
     let mut dst_mat = Mat::default();
-    imgproc::dilate(
+    imgproc::erode(
         &mat,
         &mut dst_mat,
         &kernel,
@@ -222,6 +223,18 @@ fn main() {
 
     println!("opencv exec time {:?}", exec_time.elapsed());
 
+    let new_image = morphology_image(
+        img,
+        MorphExOp::Erode,
+        &structuring_element,
+        KernelShape::new(se_size, se_size),
+        BorderMode::default(),
+        MorphologyThreadingPolicy::default(),
+    )
+    .unwrap();
+
+    new_image.save("dilated.jpg").unwrap();
+
     image::save_buffer(
         "converted.png",
         &bytes,