Update app.py
Browse files
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()
|