v3: resize-convolution upsampling — removes checkerboard artifacts

2026-06-09 07:27:14 +08:00
parent d25fd5eaa8
commit daf3d68247
12 changed files with 195 additions and 186 deletions
--- a/src/visualize.py
+++ b/src/visualize.py
@@ -34,8 +34,9 @@ def main():
    cargs = ckpt.get("args", {})
    base = cargs.get("base", 64)
    add_dist = bool(cargs.get("distance", False))
+    up_mode = cargs.get("up_mode", "deconv")
    in_ch = 3 if add_dist else 2
-    model = UNet(in_channels=in_ch, out_channels=1, base=base).to(device)
+    model = UNet(in_channels=in_ch, out_channels=1, base=base, up_mode=up_mode).to(device)
    model.load_state_dict(ckpt["model_state"])
    model.eval()
    print(f"Loaded {args.ckpt} | in_channels={in_ch} | distance={add_dist} | "