fix: inverse mask
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def extract_image(img, pos_prompts, neg_prompts, threshold):
|
|
| 52 |
pos_mask = np.any(np.stack(positive_masks), axis=0)
|
| 53 |
neg_mask = np.any(np.stack(negative_masks), axis=0)
|
| 54 |
final_mask = pos_mask & ~neg_mask
|
| 55 |
-
inverse_mask = pos_mask
|
| 56 |
|
| 57 |
# extract the final image
|
| 58 |
final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
|
|
|
|
| 52 |
pos_mask = np.any(np.stack(positive_masks), axis=0)
|
| 53 |
neg_mask = np.any(np.stack(negative_masks), axis=0)
|
| 54 |
final_mask = pos_mask & ~neg_mask
|
| 55 |
+
inverse_mask = ~pos_mask & neg_mask
|
| 56 |
|
| 57 |
# extract the final image
|
| 58 |
final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
|