danvics/immich-automated-selfie-timelapse
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Face alignment: use repeat fill instead of black fill

Browse source
This commit is contained in:
Arnaud_Cayrol 2026-02-09 20:18:41 +01:00
parent 4c98fc8d60
commit 788b4b620e
2 changed files with 29 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
src/pipeline/steps/alignment.rs
View file

@ -248,40 +248,38 @@ fn apply_affine_transform(
// Map output pixel to source pixel // Map output pixel to source pixel
let (src_x, src_y) = inv_transform.transform_point(x as f32, y as f32); let (src_x, src_y) = inv_transform.transform_point(x as f32, y as f32);
// Bilinear interpolation // Clamp coordinates to valid range (replicate edge pixels)
if src_x >= 0.0 && src_x < (width - 1) as f32 && src_y >= 0.0 && src_y < (height - 1) as f32 let clamped_x = src_x.clamp(0.0, (width - 1) as f32);
{ let clamped_y = src_y.clamp(0.0, (height - 1) as f32);
let x0 = src_x.floor() as u32;
let y0 = src_y.floor() as u32;
let x1 = x0 + 1;
let y1 = y0 + 1;
let dx = src_x - x0 as f32; // Bilinear interpolation with clamped coordinates
let dy = src_y - y0 as f32; let x0 = clamped_x.floor() as u32;
let y0 = clamped_y.floor() as u32;
let x1 = (x0 + 1).min(width - 1);
let y1 = (y0 + 1).min(height - 1);
let p00 = rgb.get_pixel(x0, y0); let dx = clamped_x - x0 as f32;
let p10 = rgb.get_pixel(x1, y0); let dy = clamped_y - y0 as f32;
let p01 = rgb.get_pixel(x0, y1);
let p11 = rgb.get_pixel(x1, y1);
let r = (p00[0] as f32 * (1.0 - dx) * (1.0 - dy) let p00 = rgb.get_pixel(x0, y0);
+ p10[0] as f32 * dx * (1.0 - dy) let p10 = rgb.get_pixel(x1, y0);
+ p01[0] as f32 * (1.0 - dx) * dy let p01 = rgb.get_pixel(x0, y1);
+ p11[0] as f32 * dx * dy) as u8; let p11 = rgb.get_pixel(x1, y1);
let g = (p00[1] as f32 * (1.0 - dx) * (1.0 - dy)
+ p10[1] as f32 * dx * (1.0 - dy)
+ p01[1] as f32 * (1.0 - dx) * dy
+ p11[1] as f32 * dx * dy) as u8;
let b = (p00[2] as f32 * (1.0 - dx) * (1.0 - dy)
+ p10[2] as f32 * dx * (1.0 - dy)
+ p01[2] as f32 * (1.0 - dx) * dy
+ p11[2] as f32 * dx * dy) as u8;
Rgb([r, g, b]) let r = (p00[0] as f32 * (1.0 - dx) * (1.0 - dy)
} else { + p10[0] as f32 * dx * (1.0 - dy)
// Out of bounds - use black + p01[0] as f32 * (1.0 - dx) * dy
Rgb([0, 0, 0]) + p11[0] as f32 * dx * dy) as u8;
} let g = (p00[1] as f32 * (1.0 - dx) * (1.0 - dy)
+ p10[1] as f32 * dx * (1.0 - dy)
+ p01[1] as f32 * (1.0 - dx) * dy
+ p11[1] as f32 * dx * dy) as u8;
let b = (p00[2] as f32 * (1.0 - dx) * (1.0 - dy)
+ p10[2] as f32 * dx * (1.0 - dy)
+ p01[2] as f32 * (1.0 - dx) * dy
+ p11[2] as f32 * dx * dy) as u8;
Rgb([r, g, b])
}); });
DynamicImage::ImageRgb8(output) DynamicImage::ImageRgb8(output)

2
src/pipeline/steps/blur.rs
View file

@ -254,7 +254,7 @@ impl ProcessingStep for BlurStep {
} }
// Draw gradient magnitude text value // Draw gradient magnitude text value
let text = format!("Grad:{:.1}", gradient_mag); let text = format!("{:.1}", gradient_mag);
draw_simple_text(&mut debug_img, 5, bar_y + 6, &text, Rgb([255, 255, 255])); draw_simple_text(&mut debug_img, 5, bar_y + 6, &text, Rgb([255, 255, 255]));
Some(DynamicImage::ImageRgb8(debug_img)) Some(DynamicImage::ImageRgb8(debug_img))