Update app.py

app.py

@@ -84,6 +84,28 @@ def predict_2(image):
 
         return out, unpatched_prediction
 
+from empatches import EMPatches
+
+def predict_patches(image):
+        #img = Image.open(image)
+        image = Image.fromarray(image).resize((size,size))
+        image = np.array(image)/255
+        emp = EMPatches()
+        img_patches, indices = emp.extract_patches(image, patchsize=pach_size, overlap=0.2, stride=int(pach_size/2))
+        pred = model.predict(np.array(img_patches))
+        # Postprocess the mask
+        #print(pred)
+        pred = np.argmax(pred, axis=3)
+        merged_img = emp.merge_patches(pred, indices, mode='max').astype("int16")
+
+        #print(merged_img)
+        unpatched_prediction = targets_classes_colors[merged_img]
+        image_classes = []
+        for i in range(targets_classes_colors.shape[0]):
+            image_classes.append(np.sum(np.all(unpatched_prediction == targets_classes_colors[i,:], axis=-1))/(size*size*3))
+        results = dict(zip(LABEL_NAMES, image_classes))
+        return results, unpatched_prediction
+
 LABEL_NAMES = ["unlabeled",    
                "paved-area",  
                      "dirt", 
@@ -194,7 +216,8 @@ with my_app:
             output_label = gr.Label(label="Predicted Masked Image")
             img_output = gr.Image(label="Image Output")
     source_image_loader.click(
-        predict_2,
+        #predict_2,
+        predict_patches,
         [
             img_source
         ],