|
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(); |
|
} |
|
|