add lables material

.gitignore

@@ -0,0 +1,32 @@
+
+**/*.DS_store
+.vscode
+node_modules
+
+__pycache__
+*.py[cod]
+
+*.egg-info
+.project
+.pydevproject
+**/*.ipynb_checkpoints
+.idea
+*.swp
+
+*~
+
+build/
+dist/
+
+/eta/
+
+/docs/build/
+/docs/source/api/
+
+/bin
+/lib
+coverage.xml
+.coverage.*
+pyvenv.cfg
+
+flagged/

app.py

@@ -6,6 +6,78 @@ from PIL import Image
 # Initialize the pipeline with your model
 pipe = pipeline("image-classification", model="SubterraAI/ofwat_material_classification")
 
+material_codes = {
+    "AC": "Asphalt Concrete",
+    "BL": "Block",
+    "BR": "Brick",
+    "CI": "Cast Iron",
+    "CO": "Concrete",
+    "CS": "Corrugated Steel",
+    "DI": "Ductile Iron",
+    "EP": "Epoxy",
+    "GI": "Galvanized Iron",
+    "MAR": "Masonry",
+    "N": "Not specified, possibly a custom abbreviation",
+    "OTH": "Other",
+    "PE": "Polyethylene",
+    "PF": "Plywood-Faced",
+    "PP": "Polypropylene",
+    "PVC": "Polyvinyl Chloride",
+    "RC": "Reinforced Concrete",
+    "ST": "Steel",
+    "U": "Unspecified, possibly a custom abbreviation",
+    "UPVC": "Unplasticized Polyvinyl Chloride",
+    "VC": "Vinyl Coated",
+    "XI": "Extra Impact",
+    "XP": "Extruded Polystyrene",
+    "Z": "Not specified, possibly a custom abbreviation"
+}
+
+material_full_names_list = [
+    'Asphalt Concrete',
+    'Block',
+    'Brick',
+    'Cast Iron',
+    'Concrete',
+    'Corrugated Steel',
+    'Ductile Iron',
+    'Epoxy',
+    'Galvanized Iron',
+    'Masonry',
+    'Not specified, possibly a custom abbreviation',
+    'Other',
+    'Polyethylene',
+    'Plywood-Faced',
+    'Polypropylene',
+    'Polyvinyl Chloride',
+    'Reinforced Concrete',
+    'Steel',
+    'Unspecified, possibly a custom abbreviation',
+    'Unplasticized Polyvinyl Chloride',
+    'Vinyl Coated',
+    'Extra Impact',
+    'Extruded Polystyrene',
+    'Not specified, possibly a custom abbreviation',
+    'Vitrified Clay Lined'
+]
+
+def replace_label_with_full_name(res, defect_dict_key_code):
+    new_res = {}
+    for dic in res:
+        # Splitting the label to handle possible suffix
+        parts = dic["label"].split('_', 1)
+        code = parts[0]
+        suffix = '_' + parts[1] if len(parts) > 1 else ''
+
+        # Replacing the code with its full name, if it exists in the dictionary
+        full_name = defect_dict_key_code.get(code, code)
+
+        # Constructing the new label with the suffix if it exists
+        new_label = full_name + suffix
+        new_res[new_label] = dic["score"]
+    
+    return new_res
+
 def classify_image(image):
     # Convert the input image to PIL format
     PIL_image = Image.fromarray(image).convert('RGB')
@@ -14,7 +86,7 @@ def classify_image(image):
     res = pipe(PIL_image)
     
     # Extract labels and scores
-    return {dic["label"]: dic["score"] for dic in res}
+    return replace_label_with_full_name(res, material_codes)
 
 
 # Create the Gradio interface
@@ -29,7 +101,9 @@ iface = gr.Interface(
         ["examples/RC.jpg"]
     ],
     description="Upload an image to classify its material.",
-    title="Material Classification with AI by Subterra"
+    title="Material Classification with AI by Subterra",
+    allow_flagging="manual",
+    flagging_options=material_full_names_list
 )
 
 # Launch the interface

flagged/log.csv

@@ -5,3 +5,4 @@ image,output 0,output 1,flag,username,timestamp
 | GI | Galvanized Iron |
 | PP | Polypropylene |
 | RC | Reinforced Concrete |",,,2023-11-27 15:26:38.548865
+"{""path"":""flagged/image/d2b68ddde05e27a540cc/PP.jpg"",""url"":""http://127.0.0.1:7860/file=/private/var/folders/5q/yl8pmxm116g6r3k8fd9gk74m0000gn/T/gradio/12885a897d79141c54ccb542582449159242eaa0/PP.jpg"",""size"":null,""orig_name"":""PP.jpg"",""mime_type"":null}","{""label"":""Vinyl Coated"",""confidences"":[{""label"":""Vinyl Coated"",""confidence"":0.7706952691078186},{""label"":""Polyvinyl Chloride"",""confidence"":0.20648105442523956},{""label"":""Polyethylene"",""confidence"":0.005161861423403025},{""label"":""Plywood-Faced"",""confidence"":0.004196972120553255},{""label"":""Concrete"",""confidence"":0.0034184197429567575}]}",Block,,2024-01-09 12:20:38.065559