add labels
Browse files
app.py
CHANGED
@@ -130,6 +130,20 @@ model = keras.models.load_model(MODEL_PATH)
|
|
130 |
images = os.listdir(IMAGE_DIR)
|
131 |
images = [f'{IMAGE_DIR}/{image}' for image in images]
|
132 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
133 |
# UserInterface
|
134 |
st.header("3rdMolar Segmentation")
|
135 |
st.subheader("Select Image:")
|
@@ -140,6 +154,7 @@ if image_file is not None:
|
|
140 |
img , output1 = segment_molar(image_file)
|
141 |
output2 = measure_molar(image_file)
|
142 |
|
|
|
143 |
st.image(img[0],width=850)
|
144 |
st.image(output1,width=850)
|
145 |
st.image(output2,width=850)
|
|
|
130 |
images = os.listdir(IMAGE_DIR)
|
131 |
images = [f'{IMAGE_DIR}/{image}' for image in images]
|
132 |
|
133 |
+
# Labels
|
134 |
+
labels = {'fatemeh_hafari_27022022_181119right.png': 'classI',
|
135 |
+
'majid_mohammadkhanlu_19022022_175338right.png': 'mesioangular',
|
136 |
+
'zahra_khodabandeh_17052022_162804right.png': 'mesioangular',
|
137 |
+
'133028019739531250right.png': 'classII',
|
138 |
+
'khadijeh_fard_aghababaei_03012022_120330right.png': 'B',
|
139 |
+
'moharam_ali_khamseh_01112021_194329right.png': 'B',
|
140 |
+
'parniya_mohamadinasab_08112021_184639right.png': 'B',
|
141 |
+
'afsaneh_khalaji_24022022_183000right.png': 'B',
|
142 |
+
'sogand_rahmani_13072022_113636right.png': 'B',
|
143 |
+
'fatemeh_mirzaei_13072022_101734right.png': 'mesioangular',
|
144 |
+
'fatemeh_hoseini_24012022_111330right.png': 'C',
|
145 |
+
'padisar_ebrahimi_13122021_111512right.png': 'mesioangular'}
|
146 |
+
|
147 |
# UserInterface
|
148 |
st.header("3rdMolar Segmentation")
|
149 |
st.subheader("Select Image:")
|
|
|
154 |
img , output1 = segment_molar(image_file)
|
155 |
output2 = measure_molar(image_file)
|
156 |
|
157 |
+
st.subheader(f"Label : {labels[image_file]}")
|
158 |
st.image(img[0],width=850)
|
159 |
st.image(output1,width=850)
|
160 |
st.image(output2,width=850)
|