quick-start SR uses showcase config (cfg 2.5 + bicubic warm start)
Browse files
README.md
CHANGED
|
@@ -89,9 +89,14 @@ dino = hf_hub_download("MVRL/genesis", "dinov3_vitl16_pretrain_sat493m-eadcf0
|
|
| 89 |
tile = fetch_tile_at_zoom(lon=151.2153, lat=-33.8568, zoom=16) # one real 256×256 tile
|
| 90 |
tile.save(f"{out}/01_original_tile.png")
|
| 91 |
|
| 92 |
-
# SR: z16 parent → 512×512 mosaic of its four z17 children
|
|
|
|
|
|
|
| 93 |
tile.save(f"{out}/02_sr_input.png")
|
| 94 |
sr_model, _ = load_sr_denoiser(sr_ckpt, "JiT-H/16", dino_weights=dino, device=device)
|
|
|
|
|
|
|
|
|
|
| 95 |
run_superresolution(sr_model, tile, device, target_zoom=17).save(f"{out}/03_sr_output.png")
|
| 96 |
|
| 97 |
# OP: keep the upper-left quadrant real, outpaint the other three (white = hole)
|
|
|
|
| 89 |
tile = fetch_tile_at_zoom(lon=151.2153, lat=-33.8568, zoom=16) # one real 256×256 tile
|
| 90 |
tile.save(f"{out}/01_original_tile.png")
|
| 91 |
|
| 92 |
+
# SR: z16 parent → 512×512 mosaic of its four z17 children.
|
| 93 |
+
# Showcase config = the paper's pyramid-engine SR settings (cfg 2.5 + bicubic
|
| 94 |
+
# warm start); the engine reads `cfg_scale`, so set the attribute (not args.cfg).
|
| 95 |
tile.save(f"{out}/02_sr_input.png")
|
| 96 |
sr_model, _ = load_sr_denoiser(sr_ckpt, "JiT-H/16", dino_weights=dino, device=device)
|
| 97 |
+
sr_model.cfg_scale = 2.5
|
| 98 |
+
sr_model.args.bicubic_init = True
|
| 99 |
+
sr_model.args.bicubic_init_t_start = 0.4
|
| 100 |
run_superresolution(sr_model, tile, device, target_zoom=17).save(f"{out}/03_sr_output.png")
|
| 101 |
|
| 102 |
# OP: keep the upper-left quadrant real, outpaint the other three (white = hole)
|