revert normalization
Browse files
app.py
CHANGED
|
@@ -28,9 +28,9 @@ def process_image(image, prompt):
|
|
| 28 |
mask = np.array(mask)[:, :, 0]
|
| 29 |
|
| 30 |
# normalize the mask
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
return mask
|
| 35 |
|
| 36 |
|
|
|
|
| 28 |
mask = np.array(mask)[:, :, 0]
|
| 29 |
|
| 30 |
# normalize the mask
|
| 31 |
+
mask_min = mask.min()
|
| 32 |
+
mask_max = mask.max()
|
| 33 |
+
mask = (mask - mask_min) / (mask_max - mask_min)
|
| 34 |
return mask
|
| 35 |
|
| 36 |
|