huggingface / candle

Minimalist ML framework for Rust

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

load_image and load_image_and_resize return different permutations

jeroenvlek opened this issue · comments

In candle-examples/src/lib.rs the function load_image returns (channels, height, width), but the function load_image_and_resize returns (channels, width, height):

let data = Tensor::from_vec(data, (height, width, 3), &Device::Cpu)?.permute((2, 0, 1))?;

and

Tensor::from_vec(data, (width, height, 3), &Device::Cpu)?.permute((2, 0, 1))