ParamAhuja commited on
Commit
aa37e25
·
1 Parent(s): 6b320c6
Files changed (1) hide show
  1. inference.py +5 -1
inference.py CHANGED
@@ -62,7 +62,11 @@ def run_inference(image, models_dict, x8_mode=False, device="cpu"):
62
  results = {}
63
 
64
  for model_name in ["srcnn", "satlas", "esrgan"]:
65
- if model_name in models_dict:
 
 
 
 
66
  try:
67
  print(f"Running inference with {model_name}...")
68
  if model_name == "srcnn":
 
62
  results = {}
63
 
64
  for model_name in ["srcnn", "satlas", "esrgan"]:
65
+ if model_name == "satlas":
66
+ print("Bypassing tiled inference for satlas backbone, using bicubic placeholder.")
67
+ w, h = image.size
68
+ sr_img = image.resize((w*4, h*4), Image.BICUBIC)
69
+ elif model_name in models_dict:
70
  try:
71
  print(f"Running inference with {model_name}...")
72
  if model_name == "srcnn":