huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Segment-anything fixes: avoid normalizing twice. (#767)

Browse Source 
* Segment-anything fixes: avoid normalizing twice.

* More fixes for the image aspect ratio.
This commit is contained in:
Laurent Mazare
2023-09-07 21:45:16 +01:00
committed by GitHub
parent 7396b8ed1a
commit 79c27fc489
3 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
candle-examples/src/lib.rs
View File

@ -16,6 +16,34 @@ pub fn device(cpu: bool) -> Result<Device> {
}
}
pub fn load_image<P: AsRef<std::path::Path>>(
p: P,
resize_longest: Option<usize>,
) -> Result<Tensor> {
let img = image::io::Reader::open(p)?
.decode()
.map_err(candle::Error::wrap)?;
let img = match resize_longest {
None => img,
Some(resize_longest) => {
let (height, width) = (img.height(), img.width());
let resize_longest = resize_longest as u32;
let (height, width) = if height < width {
let h = (resize_longest * height) / width;
(h, resize_longest)
} else {
let w = (resize_longest * width) / height;
(resize_longest, w)
};
img.resize_exact(width, height, image::imageops::FilterType::CatmullRom)
}
};
let (height, width) = (img.height() as usize, img.width() as usize);
let img = img.to_rgb8();
let data = img.into_raw();
Tensor::from_vec(data, (height, width, 3), &Device::Cpu)?.permute((2, 0, 1))
}
pub fn load_image_and_resize<P: AsRef<std::path::Path>>(
p: P,
width: usize,