Spaces:
Runtime error
Runtime error
Commit
•
a2c3378
1
Parent(s):
f928485
Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
|
@@ -36,7 +36,7 @@ def visualize_prediction(
|
|
| 36 |
new_mask = Image.fromarray(new_mask)
|
| 37 |
|
| 38 |
# Display the new mask as a semi-transparent overlay
|
| 39 |
-
ax.imshow(new_mask, alpha=0.5, cmap='
|
| 40 |
|
| 41 |
colors = COLORS * 100
|
| 42 |
for score, (xmin, ymin, xmax, ymax), label, color in zip(
|
|
|
|
| 36 |
new_mask = Image.fromarray(new_mask)
|
| 37 |
|
| 38 |
# Display the new mask as a semi-transparent overlay
|
| 39 |
+
ax.imshow(new_mask, alpha=0.5, cmap='viridis')
|
| 40 |
|
| 41 |
colors = COLORS * 100
|
| 42 |
for score, (xmin, ymin, xmax, ymax), label, color in zip(
|