diff options
Diffstat (limited to 'candle-examples/examples')
-rw-r--r-- | candle-examples/examples/beit/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/blip/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/clip/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/eva2/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/llava/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/moondream/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/segment-anything/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/stable-diffusion/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/trocr/image_processor.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/yolo-v3/main.rs | 2 | ||||
-rw-r--r-- | candle-examples/examples/yolo-v8/main.rs | 2 |
11 files changed, 11 insertions, 11 deletions
diff --git a/candle-examples/examples/beit/main.rs b/candle-examples/examples/beit/main.rs index a256fd99..47db4c66 100644 --- a/candle-examples/examples/beit/main.rs +++ b/candle-examples/examples/beit/main.rs @@ -16,7 +16,7 @@ use candle_transformers::models::beit; /// Loads an image from disk using the image crate, this returns a tensor with shape /// (3, 384, 384). Beit special normalization is applied. pub fn load_image384_beit_norm<P: AsRef<std::path::Path>>(p: P) -> Result<Tensor> { - let img = image::io::Reader::open(p)? + let img = image::ImageReader::open(p)? .decode() .map_err(candle::Error::wrap)? .resize_to_fill(384, 384, image::imageops::FilterType::Triangle); diff --git a/candle-examples/examples/blip/main.rs b/candle-examples/examples/blip/main.rs index 15e36476..d971b49d 100644 --- a/candle-examples/examples/blip/main.rs +++ b/candle-examples/examples/blip/main.rs @@ -55,7 +55,7 @@ const SEP_TOKEN_ID: u32 = 102; /// Loads an image from disk using the image crate, this returns a tensor with shape /// (3, 384, 384). OpenAI normalization is applied. pub fn load_image<P: AsRef<std::path::Path>>(p: P) -> Result<Tensor> { - let img = image::io::Reader::open(p)? + let img = image::ImageReader::open(p)? .decode() .map_err(candle::Error::wrap)? .resize_to_fill(384, 384, image::imageops::FilterType::Triangle); diff --git a/candle-examples/examples/clip/main.rs b/candle-examples/examples/clip/main.rs index f301d211..d057663d 100644 --- a/candle-examples/examples/clip/main.rs +++ b/candle-examples/examples/clip/main.rs @@ -33,7 +33,7 @@ struct Args { } fn load_image<T: AsRef<std::path::Path>>(path: T, image_size: usize) -> anyhow::Result<Tensor> { - let img = image::io::Reader::open(path)?.decode()?; + let img = image::ImageReader::open(path)?.decode()?; let (height, width) = (image_size, image_size); let img = img.resize_to_fill( width as u32, diff --git a/candle-examples/examples/eva2/main.rs b/candle-examples/examples/eva2/main.rs index 4270075d..1a3a82cc 100644 --- a/candle-examples/examples/eva2/main.rs +++ b/candle-examples/examples/eva2/main.rs @@ -16,7 +16,7 @@ use candle_transformers::models::eva2; /// Loads an image from disk using the image crate, this returns a tensor with shape /// (3, 448, 448). OpenAI normalization is applied. pub fn load_image448_openai_norm<P: AsRef<std::path::Path>>(p: P) -> Result<Tensor> { - let img = image::io::Reader::open(p)? + let img = image::ImageReader::open(p)? .decode() .map_err(candle::Error::wrap)? .resize_to_fill(448, 448, image::imageops::FilterType::Triangle); diff --git a/candle-examples/examples/llava/main.rs b/candle-examples/examples/llava/main.rs index d6c911af..cb809300 100644 --- a/candle-examples/examples/llava/main.rs +++ b/candle-examples/examples/llava/main.rs @@ -57,7 +57,7 @@ fn load_image<T: AsRef<std::path::Path>>( llava_config: &LLaVAConfig, dtype: DType, ) -> Result<((u32, u32), Tensor)> { - let img = image::io::Reader::open(path)?.decode()?; + let img = image::ImageReader::open(path)?.decode()?; let img_tensor = process_image(&img, processor, llava_config)?; Ok(((img.width(), img.height()), img_tensor.to_dtype(dtype)?)) } diff --git a/candle-examples/examples/moondream/main.rs b/candle-examples/examples/moondream/main.rs index a3dc67ee..6e099888 100644 --- a/candle-examples/examples/moondream/main.rs +++ b/candle-examples/examples/moondream/main.rs @@ -208,7 +208,7 @@ struct Args { /// Loads an image from disk using the image crate, this returns a tensor with shape /// (3, 378, 378). pub fn load_image<P: AsRef<std::path::Path>>(p: P) -> candle::Result<Tensor> { - let img = image::io::Reader::open(p)? + let img = image::ImageReader::open(p)? .decode() .map_err(candle::Error::wrap)? .resize_to_fill(378, 378, image::imageops::FilterType::Triangle); // Adjusted to 378x378 diff --git a/candle-examples/examples/segment-anything/main.rs b/candle-examples/examples/segment-anything/main.rs index 10f65c66..204575da 100644 --- a/candle-examples/examples/segment-anything/main.rs +++ b/candle-examples/examples/segment-anything/main.rs @@ -139,7 +139,7 @@ pub fn main() -> anyhow::Result<()> { let (_one, h, w) = mask.dims3()?; let mask = mask.expand((3, h, w))?; - let mut img = image::io::Reader::open(&args.image)? + let mut img = image::ImageReader::open(&args.image)? .decode() .map_err(candle::Error::wrap)?; let mask_pixels = mask.permute((1, 2, 0))?.flatten_all()?.to_vec1::<u8>()?; diff --git a/candle-examples/examples/stable-diffusion/main.rs b/candle-examples/examples/stable-diffusion/main.rs index d424444b..b6585afa 100644 --- a/candle-examples/examples/stable-diffusion/main.rs +++ b/candle-examples/examples/stable-diffusion/main.rs @@ -380,7 +380,7 @@ fn text_embeddings( } fn image_preprocess<T: AsRef<std::path::Path>>(path: T) -> anyhow::Result<Tensor> { - let img = image::io::Reader::open(path)?.decode()?; + let img = image::ImageReader::open(path)?.decode()?; let (height, width) = (img.height() as usize, img.width() as usize); let height = height - height % 32; let width = width - width % 32; diff --git a/candle-examples/examples/trocr/image_processor.rs b/candle-examples/examples/trocr/image_processor.rs index 531caa56..3571d6d3 100644 --- a/candle-examples/examples/trocr/image_processor.rs +++ b/candle-examples/examples/trocr/image_processor.rs @@ -145,7 +145,7 @@ impl ViTImageProcessor { pub fn load_images(&self, image_path: Vec<&str>) -> Result<Vec<image::DynamicImage>> { let mut images: Vec<image::DynamicImage> = Vec::new(); for path in image_path { - let img = image::io::Reader::open(path)?.decode().unwrap(); + let img = image::ImageReader::open(path)?.decode().unwrap(); images.push(img); } diff --git a/candle-examples/examples/yolo-v3/main.rs b/candle-examples/examples/yolo-v3/main.rs index a6574697..fb46dac2 100644 --- a/candle-examples/examples/yolo-v3/main.rs +++ b/candle-examples/examples/yolo-v3/main.rs @@ -159,7 +159,7 @@ pub fn main() -> Result<()> { let net_width = darknet.width()?; let net_height = darknet.height()?; - let original_image = image::io::Reader::open(&image_name)? + let original_image = image::ImageReader::open(&image_name)? .decode() .map_err(candle::Error::wrap)?; let image = { diff --git a/candle-examples/examples/yolo-v8/main.rs b/candle-examples/examples/yolo-v8/main.rs index eb338647..084a42d5 100644 --- a/candle-examples/examples/yolo-v8/main.rs +++ b/candle-examples/examples/yolo-v8/main.rs @@ -390,7 +390,7 @@ pub fn run<T: Task>(args: Args) -> anyhow::Result<()> { for image_name in args.images.iter() { println!("processing {image_name}"); let mut image_name = std::path::PathBuf::from(image_name); - let original_image = image::io::Reader::open(&image_name)? + let original_image = image::ImageReader::open(&image_name)? .decode() .map_err(candle::Error::wrap)?; let (width, height) = { |