SMD00 commited on
Commit
74d5983
1 Parent(s): 6949412

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -301,7 +301,7 @@ def visualize(model, data, dims):
301
  real_imgs = lab_to_rgb(L, real_color)
302
  # img=cv.resize(fake_imgs[0], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC)
303
  img=fake_imgs[0]
304
- return img.resize((dims[1], dims[0]))
305
  return fake_imgs[0]
306
  for i in range(1):
307
  # t_img = transforms.Resize((dims[0], dims[1]))(t_img)
 
301
  real_imgs = lab_to_rgb(L, real_color)
302
  # img=cv.resize(fake_imgs[0], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC)
303
  img=fake_imgs[0]
304
+ return np.resize(img,(dims[1], dims[0]))
305
  return fake_imgs[0]
306
  for i in range(1):
307
  # t_img = transforms.Resize((dims[0], dims[1]))(t_img)