kornia-rs
kornia-rs copied to clipboard
Memory leak in `remap`
There is some amount of memory being allocated during the remap
function that isn't getting cleaned up.
This is with the latest commit on main. So I imagine it's something to do with rayon or the new tensor implementation.
This simple program will slowly grow in memory usage:
fn main() {
let size = ImageSize {
width: 1440,
height: 1440,
};
let correction_map = correction_map(&size);
loop {
let input = Image::<u8, 3>::from_size_val(size, 0).unwrap();
let mut output = Image::<f32, 3>::from_size_val(size, 0.0).unwrap();
remap(
&input.cast().unwrap(),
&mut output,
&correction_map.0,
&correction_map.1,
InterpolationMode::Bilinear,
)
.unwrap();
}
}