mdeville/remove_background
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

png as default output

Browse source
This commit is contained in:
Matthew Deville 2026-04-23 14:26:13 +02:00
parent 31d84c99d7
commit 3be68fcccf
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/main.rs
View file

@ -163,7 +163,8 @@ fn main() -> Result<(), Box<dyn Error>> {
let mask = session.predict(&img)?; let mask = session.predict(&img)?;
let result_rgba = apply_mask(&img, &mask)?; let result_rgba = apply_mask(&img, &mask)?;
let output_path = output_directory.join(path.file_name().unwrap()); let mut output_path = output_directory.join(path.file_name().unwrap());
output_path.set_extension("png");
result_rgba.save(&output_path)?; result_rgba.save(&output_path)?;
processed += 1; processed += 1;
info!(output = %output_path.display(), "Processed image saved"); info!(output = %output_path.display(), "Processed image saved");