waifu-2x / src /main.rs
Aster
Add test codes
68cea2d
raw
history blame contribute delete
No virus
549 Bytes
use std::path::PathBuf;
use image::io::Reader;
use waifu2x::{Convolution7, DenoiseLevel, Environment};
fn main() {
let here = PathBuf::from(".").canonicalize().unwrap();
println!("here: {}", here.display());
let env = Environment::default().into_arc();
let mut model = Convolution7::new(&env, &PathBuf::from("models/upconv7"), DenoiseLevel::High).unwrap();
let image = Reader::open("tests/miku.png").unwrap().decode().unwrap();
let out = model.upscale_2x(&image).unwrap();
out.save("tests/miku_conv7.png").unwrap();
}