Commit
·
61a9d1f
1
Parent(s):
5e179d4
change to tuple
Browse files
app.py
CHANGED
|
@@ -160,9 +160,9 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
|
|
| 160 |
# get unique RGB colors from this segmentation map
|
| 161 |
st.image(Image.fromarray(real_seg), width=512)
|
| 162 |
print(real_seg.shape)
|
| 163 |
-
print(real_seg)
|
| 164 |
unique_colors = np.unique(real_seg.reshape(-1, real_seg.shape[2]), axis=0)
|
| 165 |
print(unique_colors)
|
|
|
|
| 166 |
st.session_state['unique_colors'] = unique_colors
|
| 167 |
# multiselect to choose multiple colors
|
| 168 |
chosen_colors = st.multiselect(
|
|
|
|
| 160 |
# get unique RGB colors from this segmentation map
|
| 161 |
st.image(Image.fromarray(real_seg), width=512)
|
| 162 |
print(real_seg.shape)
|
|
|
|
| 163 |
unique_colors = np.unique(real_seg.reshape(-1, real_seg.shape[2]), axis=0)
|
| 164 |
print(unique_colors)
|
| 165 |
+
unique_colors = (tuple(color) for color in unique_colors)
|
| 166 |
st.session_state['unique_colors'] = unique_colors
|
| 167 |
# multiselect to choose multiple colors
|
| 168 |
chosen_colors = st.multiselect(
|