changed both alpha in output
Browse files
output.py
CHANGED
@@ -36,9 +36,9 @@ def visualize_output(image, output, binarize, threshold):
|
|
36 |
plt.axis('off')
|
37 |
plt.imshow(image)
|
38 |
if binarize:
|
39 |
-
plt.imshow(output_mask, alpha=.
|
40 |
else:
|
41 |
-
plt.imshow(output_mask, alpha=.
|
42 |
fig.tight_layout(pad=0)
|
43 |
fig.canvas.draw()
|
44 |
data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
|
|
|
36 |
plt.axis('off')
|
37 |
plt.imshow(image)
|
38 |
if binarize:
|
39 |
+
plt.imshow(output_mask, alpha=.4)
|
40 |
else:
|
41 |
+
plt.imshow(output_mask, alpha=.4)
|
42 |
fig.tight_layout(pad=0)
|
43 |
fig.canvas.draw()
|
44 |
data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
|