Spaces:
Build error
Build error
nounouille commited on
Commit ·
5ae7114
1
Parent(s): a0ce4c5
� Revert : retour à version stable sans export CSV
Browse files
app.py
CHANGED
|
@@ -40,12 +40,8 @@ def generate_dummy_legend():
|
|
| 40 |
|
| 41 |
return img
|
| 42 |
|
| 43 |
-
# legend_img = Image.new("RGB", (300, 50), color=(128, 128, 128))
|
| 44 |
legend_img = generate_dummy_legend()
|
| 45 |
history = deque(maxlen=5)
|
| 46 |
-
# history_csv = []
|
| 47 |
-
|
| 48 |
-
|
| 49 |
|
| 50 |
def overlay_mask_on_image(image: Image.Image, mask: Image.Image, alpha: float = 0.5) -> Image.Image:
|
| 51 |
image = image.convert("RGBA").resize(mask.size)
|
|
@@ -69,12 +65,6 @@ def segment_image(image: Image.Image, source_name="image_upload.png"):
|
|
| 69 |
mask_image = Image.open(io.BytesIO(mask_bytes)).convert("RGB")
|
| 70 |
overlay_img = overlay_mask_on_image(image, mask_image)
|
| 71 |
|
| 72 |
-
# # Ajout dans l'historique CSV
|
| 73 |
-
# history_csv.append({
|
| 74 |
-
# "filename": source_name,
|
| 75 |
-
# "inference_time": round(float(data["inference_time"]), 3)
|
| 76 |
-
# })
|
| 77 |
-
|
| 78 |
return image, mask_image, overlay_img, f"{data['inference_time']} sec"
|
| 79 |
|
| 80 |
except RequestException as e:
|
|
@@ -104,16 +94,6 @@ def load_image_from_url(filename):
|
|
| 104 |
print(f"[ERREUR CHARGEMENT IMAGE {filename}] {e}")
|
| 105 |
return None
|
| 106 |
|
| 107 |
-
# def export_csv():
|
| 108 |
-
# if not history_csv:
|
| 109 |
-
# return gr.File.update(value=None, visible=False)
|
| 110 |
-
|
| 111 |
-
# df = pd.DataFrame(history_csv)
|
| 112 |
-
# tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode='w', encoding='utf-8')
|
| 113 |
-
# df.to_csv(tmp_file.name, index=False)
|
| 114 |
-
# tmp_file.close()
|
| 115 |
-
|
| 116 |
-
# return gr.File.update(value=tmp_file.name, visible=True, label="📥 Résultats CSV")
|
| 117 |
|
| 118 |
with gr.Blocks(title="Segmentation d'Images Urbaines") as demo:
|
| 119 |
gr.Markdown("# 🧠 Segmentation d'Images Urbaines")
|
|
@@ -137,10 +117,6 @@ with gr.Blocks(title="Segmentation d'Images Urbaines") as demo:
|
|
| 137 |
btn_url.click(fn=lambda name: segment_image(load_image_from_url(name), source_name=name),
|
| 138 |
inputs=dropdown, outputs=[img_original, img_mask, img_overlay, inf_time])
|
| 139 |
|
| 140 |
-
# with gr.Row():
|
| 141 |
-
# btn_export = gr.Button("📤 Exporter CSV des inférences")
|
| 142 |
-
# file_output = gr.File(visible=False)
|
| 143 |
-
# btn_export.click(fn=export_csv, inputs=[], outputs=file_output)
|
| 144 |
|
| 145 |
with gr.Accordion("🎨 Légende des classes", open=False):
|
| 146 |
gr.Image(value=legend_img, label="Légende (fictive ici)", interactive=False)
|
|
|
|
| 40 |
|
| 41 |
return img
|
| 42 |
|
|
|
|
| 43 |
legend_img = generate_dummy_legend()
|
| 44 |
history = deque(maxlen=5)
|
|
|
|
|
|
|
|
|
|
| 45 |
|
| 46 |
def overlay_mask_on_image(image: Image.Image, mask: Image.Image, alpha: float = 0.5) -> Image.Image:
|
| 47 |
image = image.convert("RGBA").resize(mask.size)
|
|
|
|
| 65 |
mask_image = Image.open(io.BytesIO(mask_bytes)).convert("RGB")
|
| 66 |
overlay_img = overlay_mask_on_image(image, mask_image)
|
| 67 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
return image, mask_image, overlay_img, f"{data['inference_time']} sec"
|
| 69 |
|
| 70 |
except RequestException as e:
|
|
|
|
| 94 |
print(f"[ERREUR CHARGEMENT IMAGE {filename}] {e}")
|
| 95 |
return None
|
| 96 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 97 |
|
| 98 |
with gr.Blocks(title="Segmentation d'Images Urbaines") as demo:
|
| 99 |
gr.Markdown("# 🧠 Segmentation d'Images Urbaines")
|
|
|
|
| 117 |
btn_url.click(fn=lambda name: segment_image(load_image_from_url(name), source_name=name),
|
| 118 |
inputs=dropdown, outputs=[img_original, img_mask, img_overlay, inf_time])
|
| 119 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 120 |
|
| 121 |
with gr.Accordion("🎨 Légende des classes", open=False):
|
| 122 |
gr.Image(value=legend_img, label="Légende (fictive ici)", interactive=False)
|