diff options
Diffstat (limited to 'src/ocr.rs')
-rw-r--r-- | src/ocr.rs | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -22,12 +22,12 @@ fn x_matches(image: &GrayImage, template: &GrayImage) -> Vec<u32> { ); match_values .enumerate_pixels() - .filter(|(_x, _y, pix)| pix.0[0] > 0.9) + .filter(|(_x, _y, pix)| pix.0[0] > 0.95) .map(|(x, _y, _pix)| x) .collect() } -pub fn ocr(image: DynamicImage) -> u32 { +pub fn ocr(image: DynamicImage) -> Option<u32> { let grayscale_image = image::imageops::grayscale(&image); let mut digit_x_positions: Vec<(u8, u32)> = (0..10) .flat_map(|i| { @@ -46,5 +46,5 @@ pub fn ocr(image: DynamicImage) -> u32 { .map(|(i, _x)| format!("{}", i)) .collect(); dbg!(&digits); - digits.parse().unwrap() + digits.parse().ok() } |