Karin0616
commited on
Commit
·
59255df
1
Parent(s):
dc50072
change axis
Browse files
app.py
CHANGED
@@ -69,12 +69,14 @@ def draw_plot(pred_img, seg):
|
|
69 |
FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)
|
70 |
|
71 |
unique_labels = np.unique(seg.numpy().astype("uint8"))
|
|
|
72 |
ax = plt.subplot(grid_spec[1])
|
73 |
plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
|
74 |
-
|
75 |
-
|
76 |
-
plt.xticks(
|
77 |
-
|
|
|
78 |
return fig
|
79 |
|
80 |
def sepia(input_img):
|
|
|
69 |
FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)
|
70 |
|
71 |
unique_labels = np.unique(seg.numpy().astype("uint8"))
|
72 |
+
|
73 |
ax = plt.subplot(grid_spec[1])
|
74 |
plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
|
75 |
+
|
76 |
+
ax.xaxis.tick_bottom()
|
77 |
+
plt.xticks(range(len(unique_labels)), LABEL_NAMES[unique_labels])
|
78 |
+
plt.yticks([], [])
|
79 |
+
ax.tick_params(width=0.0, labelsize=28)
|
80 |
return fig
|
81 |
|
82 |
def sepia(input_img):
|