File size: 503 Bytes
e7c49dc | 1 2 3 4 5 6 7 8 9 10 11 | """Minimal example: denoise a photograph with İvme-Denoise-v1."""
import numpy as np
from PIL import Image
from modeling_ivmetron import load_model, denoise_tiled, estimate_sigma_map
model = load_model(".", device="cuda")
img = np.asarray(Image.open("photo.jpg").convert("RGB"), dtype=np.float32) / 255.0
out = denoise_tiled(model, img, estimate_sigma_map(img), device="cuda")
Image.fromarray((np.clip(out, 0, 1) * 255).round().astype(np.uint8)).save("clean.png", quality=95)
print("wrote clean.png")
|