fix: fix split images with filter channel issue

4o3F · Jan 15, 2025 · d611adf · d611adf
1 parent 1c8bda9
commit d611adf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/common/augment.rs b/src/common/augment.rs
@@ -719,9 +719,11 @@ pub async fn split_images_with_filter(
             let _ = permit.acquire().await.unwrap();
             tracing::info!("Processing {}", entry.file_name().unwrap().to_str().unwrap());
             let label_id = entry.file_stem().unwrap().to_str().unwrap().to_string();
-            let img = imgcodecs::imread(entry.to_str().unwrap(), imgcodecs::IMREAD_UNCHANGED)
+            let img = imgcodecs::imread(entry.to_str().unwrap(), imgcodecs::IMREAD_COLOR)
                 .unwrap();
 
+
+
             let size = img.size().unwrap();
             let (width, height) = (size.width, size.height);
             let y_count = height / target_height as i32;