.

app.py

@@ -43,22 +43,21 @@ from skimage import exposure
 import pandas as pd
 
 
-cdl_color_map = [{'value': 1, 'label': 'Natural vegetation', 'rgb': (233,255,190)},
-                 {'value': 2, 'label': 'Forest', 'rgb': (149,206,147)},
-                 {'value': 3, 'label': 'Corn', 'rgb': (255,212,0)},
-                 {'value': 4, 'label': 'Soybeans', 'rgb': (38,115,0)},
-                 {'value': 5, 'label': 'Wetlands', 'rgb': (128,179,179)},
-                 {'value': 6, 'label': 'Developed/Barren', 'rgb': (156,156,156)},
-                 {'value': 7, 'label': 'Open Water', 'rgb': (77,112,163)},
-                 {'value': 8, 'label': 'Winter Wheat', 'rgb': (168,112,0)},
-                 {'value': 9, 'label': 'Alfalfa', 'rgb': (255,168,227)},
-                 {'value': 10, 'label': 'Fallow/Idle cropland', 'rgb': (191,191,122)},
-                 {'value': 11, 'label': 'Cotton', 'rgb':(255,38,38)},
-                 {'value': 12, 'label': 'Sorghum', 'rgb':(255,158,15)},
-                 {'value': 13, 'label': 'Other', 'rgb':(0,175,77)}]
-
-
-def apply_color_map(rgb, color_map=cdl_color_map):
+cdl_color_map = [{'value': 1, 'label': 'Natural vegetation', 'rgb': (233,255,190), 'qtt': 0},
+                 {'value': 2, 'label': 'Forest', 'rgb': (149,206,147), 'qtt': 0},
+                 {'value': 3, 'label': 'Corn', 'rgb': (255,212,0), 'qtt': 0},
+                 {'value': 4, 'label': 'Soybeans', 'rgb': (38,115,0), 'qtt': 0},
+                 {'value': 5, 'label': 'Wetlands', 'rgb': (128,179,179), 'qtt': 0},
+                 {'value': 6, 'label': 'Developed/Barren', 'rgb': (156,156,156), 'qtt': 0},
+                 {'value': 7, 'label': 'Open Water', 'rgb': (77,112,163), 'qtt': 0},
+                 {'value': 8, 'label': 'Winter Wheat', 'rgb': (168,112,0), 'qtt': 0},
+                 {'value': 9, 'label': 'Alfalfa', 'rgb': (255,168,227), 'qtt': 0},
+                 {'value': 10, 'label': 'Fallow/Idle cropland', 'rgb': (191,191,122), 'qtt': 0},
+                 {'value': 11, 'label': 'Cotton', 'rgb':(255,38,38), 'qtt': 0},
+                 {'value': 12, 'label': 'Sorghum', 'rgb':(255,158,15), 'qtt': 0},
+                 {'value': 13, 'label': 'Other', 'rgb':(0,175,77), 'qtt': 0}]
+
+def apply_color_map(rgb, color_map=cdl_color_map, map_resume):
     
     rgb_mapped = rgb.copy()
     
@@ -69,7 +68,9 @@ def apply_color_map(rgb, color_map=cdl_color_map):
             cont = np.where((rgb[0] == map_tmp['value']) & (rgb[1] == map_tmp['value']) & (rgb[2] == map_tmp['value']), 1, 0)
     
         print(map_tmp['label'])
-        print(cont)
+        print("Cantidad total ", sum(cont))
+        map_resume.append({"label": map_tmp['label'],'qtt': sum(cont)})
+        
     return rgb_mapped    
     
 
@@ -179,7 +180,7 @@ def process_rgb(input, mask, indexes):
 
     return rgb
 
-def inference_on_file(target_image, model, custom_test_pipeline):
+def inference_on_file(target_image, model, custom_test_pipeline, map_resume):
 
     target_image = target_image.name
     time_taken=-1
@@ -212,8 +213,8 @@ def inference_on_file(target_image, model, custom_test_pipeline):
     output=result[0][0] + 1
     output = np.vstack([output[None], output[None], output[None]]).astype(np.uint8)
     
-
-    output=apply_color_map(output).transpose((1,2,0))
+    
+    output=apply_color_map(output).transpose((1,2,0),map_resume)
         
     return rgb1,rgb2,rgb3,output
 
@@ -243,16 +244,10 @@ config.model.backbone.pretrained=None
 model = init_segmentor(config, ckpt, device='cpu')
 custom_test_pipeline=process_test_pipeline(model.cfg.data.test.pipeline, None)
 
-func = partial(inference_on_file, model=model, custom_test_pipeline=custom_test_pipeline)
-
-
+map_resume = []
+func = partial(inference_on_file, model=model, custom_test_pipeline=custom_test_pipeline, map_resume=map_resume)
 
-simple = pd.DataFrame(
-    {
-        "a": ["A", "B", "C", "D", "E", "F", "G", "H", "I"],
-        "b": [28, 55, 43, 91, 81, 53, 19, 87, 52],
-    }
-)
+bar_data = pd.DataFrame(map_resume)
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -277,15 +272,14 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            gr.BarPlot(simple,
-                    x="a",
-                    y="b",
+            gr.BarPlot(bar_data,
+                    x="label",
+                    y="qtt",
                     title="Simple Bar Plot with made up data",
-                    tooltip=["a", "b"],
-                    y_lim=[20, 100],)
-            gr.LinePlot(simple,
-                        x='a',
-                        y='b')
+                    tooltip=["label", "Qtt"])
+            gr.LinePlot(bar_data,
+                        x='label',
+                        y='qtt')
         with gr.Column():
             out = gr.Image(image_mode='RGB', scale=10, label='Model prediction')
             # gr.Image(value='Legend.png', image_mode='RGB', scale=2, show_label=False)