basso4 commited on
Commit
054cfac
Β·
verified Β·
1 Parent(s): e606b90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -180,6 +180,7 @@ if __name__ == "__main__":
180
  dict = dict['background'].convert("RGB").resize((384, 512))
181
  dict = np.array(dict)
182
  dict = cv2.cvtColor(dict, cv2.COLOR_RGB2BGR)
 
183
 
184
  #img_output
185
  img_cv = rearrange(x_checked_image_torch[0], 'c h w -> h w c').cpu().numpy()
 
180
  dict = dict['background'].convert("RGB").resize((384, 512))
181
  dict = np.array(dict)
182
  dict = cv2.cvtColor(dict, cv2.COLOR_RGB2BGR)
183
+ dict = dict.astype(np.uint8)
184
 
185
  #img_output
186
  img_cv = rearrange(x_checked_image_torch[0], 'c h w -> h w c').cpu().numpy()