28 lines
1.2 KiB
Rust
28 lines
1.2 KiB
Rust
|
use aha::utils::interpolate::interpolate_nearest_1d;
|
||
|
|
use anyhow::Result;
|
||
|
|
use candle_core::Tensor;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn interpolate_test() -> Result<()> {
|
||
|
|
// RUST_BACKTRACE=1 cargo test -F cuda --test interpolate interpolate_test -r -- --nocapture
|
||
|
|
let device = &candle_core::Device::Cpu;
|
||
|
|
let input = Tensor::arange(0.0f32, 10.0f32, device)?;
|
||
|
|
let input_1d = input.reshape((1, 1, 10))?;
|
||
|
|
println!("input_1d: {}", input_1d);
|
||
|
|
let x_nearest_1d = interpolate_nearest_1d(&input_1d, 20)?;
|
||
|
|
println!("x_nearest_1d: {}", x_nearest_1d);
|
||
|
|
// let x_linear_1d = interpolate_linear_1d(&input_1d, 10, Some(true))?;
|
||
|
|
// println!("x_linear_1d: {}", x_linear_1d);
|
||
|
|
// let input_2d = input.reshape((1, 1, 10, 10))?;
|
||
|
|
// println!("input_2d: {}", input_2d);
|
||
|
|
// let x_nearest_2d = interpolate_nearest_2d(&input_2d, (10, 10))?;
|
||
|
|
// println!("x_nearest_2d: {}", x_nearest_2d);
|
||
|
|
// let x_bilinear = interpolate_bilinear(&input_2d, (5, 5), Some(true), Some(false))?;
|
||
|
|
// println!("x_bilinear: {}", x_bilinear);
|
||
|
|
// let x_bicubic = interpolate_bicubic(&input_2d, (5, 5), Some(false), Some(true))?;
|
||
|
|
// // let x_bicubic = interpolate_bicubic_standard(&input_2d, (5, 5), None)?;
|
||
|
|
|
||
|
|
// println!("x_bicubic: {}", x_bicubic);
|
||
|
|
Ok(())
|
||
|
|
}
|