lyndonzheng commited on
Commit
ae2071d
β€’
1 Parent(s): bffa280

update bugs

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -38,27 +38,28 @@ def main():
38
  raise gr.Error("No image uploaded!")
39
 
40
  def preprocess(image, dynamic_size=False, padding=True):
41
- w, h = image.size
42
  size = 32
43
  if dynamic_size:
44
  while max(h, w) // size > 20:
45
  size *=2
46
- image = TTF.center_crop(image, (h // size * size, w // size * size))
47
- image = TTF.resize(image, (h // size * 32, w // size * 32), interpolation=TT.InterpolationMode.BICUBIC)
48
- model.cfg.dataset.width, model.cfg.dataset.height = image.size
49
- model.set_backproject()
50
  else:
51
  model.cfg.dataset.height, model.cfg.dataset.width = 256, 384
52
- image = TTF.resize(
53
  image, (model.cfg.dataset.height, model.cfg.dataset.width),
54
  interpolation=TT.InterpolationMode.BICUBIC
55
  )
56
  if padding:
57
- image = pad_border_fn(image)
58
  model.cfg.dataset.pad_border_aug = 32
59
  else:
 
60
  model.cfg.dataset.pad_border_aug = 0
61
- return image
 
62
 
63
  @spaces.GPU()
64
  def reconstruct_and_export(image):
 
38
  raise gr.Error("No image uploaded!")
39
 
40
  def preprocess(image, dynamic_size=False, padding=True):
41
+ h, w = image.size
42
  size = 32
43
  if dynamic_size:
44
  while max(h, w) // size > 20:
45
  size *=2
46
+ crop_image = TTF.center_crop(image, (w // size * size, h // size * size))
47
+ resize_image = TTF.resize(crop_image, (w // size * 32, h // size * 32), interpolation=TT.InterpolationMode.BICUBIC)
48
+ model.cfg.dataset.width, model.cfg.dataset.height = resize_image.size
 
49
  else:
50
  model.cfg.dataset.height, model.cfg.dataset.width = 256, 384
51
+ resize_image = TTF.resize(
52
  image, (model.cfg.dataset.height, model.cfg.dataset.width),
53
  interpolation=TT.InterpolationMode.BICUBIC
54
  )
55
  if padding:
56
+ input_image = pad_border_fn(resize_image)
57
  model.cfg.dataset.pad_border_aug = 32
58
  else:
59
+ input_image = resize_image
60
  model.cfg.dataset.pad_border_aug = 0
61
+ model.set_backproject()
62
+ return input_image
63
 
64
  @spaces.GPU()
65
  def reconstruct_and_export(image):