v-1
Browse files
app.py
CHANGED
@@ -83,10 +83,10 @@ def UNet2_with_STN():
|
|
83 |
model = Model(inputs, outputs)
|
84 |
return model
|
85 |
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
|
91 |
def cellsegmentor(img):
|
92 |
|
|
|
83 |
model = Model(inputs, outputs)
|
84 |
return model
|
85 |
|
86 |
+
model = UNet2_with_STN()
|
87 |
+
mean_iou = MeanIoU(num_classes=2)
|
88 |
+
model.compile(optimizer="adam", loss="binary_crossentropy", metrics=["acc",mean_iou])
|
89 |
+
model.load_weights('./model_file/UNet+stn_model_longrun.h5')
|
90 |
|
91 |
def cellsegmentor(img):
|
92 |
|