huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

dinov2 - read images from disk and compute the class probabilities (#503)

Browse Source 
* Load the image from disk and convert it to a tensor.

* Tweak the function name.
This commit is contained in:
Laurent Mazare
2023-08-18 15:50:33 +01:00
committed by GitHub
parent 95462c6a2e
commit 4f1541526c
4 changed files with 48 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/stable-diffusion/main.rs
View File

@ -332,7 +332,7 @@ fn run(args: Args) -> Result<()> {
let image = (image * 255.)?.to_dtype(DType::U8)?.i(0)?;
let image_filename =
output_filename(&final_image, idx + 1, num_samples, Some(timestep_index + 1));
crate::utils::save_image(&image, image_filename)?
candle_examples::save_image(&image, image_filename)?
}
}
@ -346,7 +346,7 @@ fn run(args: Args) -> Result<()> {
let image = ((image / 2.)? + 0.5)?.to_device(&Device::Cpu)?;
let image = (image * 255.)?.to_dtype(DType::U8)?.i(0)?;
let image_filename = output_filename(&final_image, idx + 1, num_samples, None);
crate::utils::save_image(&image, image_filename)?
candle_examples::save_image(&image, image_filename)?
}
Ok(())
}