| import os |
| import sys |
| sys.path.append(os.getcwd()) |
|
|
| from Lib.Const import COLOR_MAP, LABELS |
| from Lib.DetectFaultOnnx import DetectFaults |
|
|
| import cv2 |
| import gradio as gr |
|
|
| demoImages = [ |
| "data/DJI_20240905095004_0007_W.JPG", |
| "data/DJI_20240905091530_0003_W.JPG", |
| "data/DJI_20240905094647_0003_W.JPG", |
| "data/DJI_20240905094647_0003_Z.JPG", |
| "data/DJI_20240905101846_0005_W.JPG", |
| "data/16_3450.png", |
| "data/16_3735.png", |
| "data/16_3900.png", |
| "data/19_00350.png", |
| "data/25_00272.png", |
| "data/67_02661.png" |
| ] |
|
|
|
|
| def Warning(): |
| gr.Info("DGH ARGE YAZILIM DANIŞMANLIK ENERJİ İNŞAAT SAN.TİC.LTD.ŞTİ", duration=0.5) |
|
|
| with gr.Blocks(css="footer{display:none !important}") as block: |
| gr.Markdown("## Yüksek Gerilim Hattı İzolatörlerinin Arıza Tespiti - Demo") |
| gr.Markdown("**Ark İzi, Kırık ve Eksik İzolatör Hatalarını Tespit Eder**") |
| with gr.Row(): |
| with gr.Column(): |
| inputImage = gr.Image(label="Fotoğraf") |
|
|
| with gr.Column(): |
| thresholdSlider = gr.Slider(0, 1, value=0.25, label="Model Eşik Değeri", info="0 ve 1 arası seçiniz.") |
| iouThresholdSlider = gr.Slider(0, 1, value=0.45, label="IOU (Intersection Over Union) Eşik Değeri", info="0 ve 1 arası seçiniz.") |
| with gr.Accordion("Demo Görsellerden Seçebilirsiniz", open=False): |
| imageGallery = gr.Examples( |
| examples=[ |
| os.path.join("data", img_name) for img_name in sorted(os.listdir("data")) |
| ], |
| inputs=[inputImage], |
| label="Örnekler", |
| cache_examples=False, |
| examples_per_page=7 |
| ) |
| processButton = gr.Button("Tespit Et") |
|
|
| results = gr.Textbox(label="Log") |
| gr.HTML("</hr>") |
| processedImageGallery = gr.Gallery( |
| label="Sonuçlar", |
| rows=1, |
| columns=2, |
| object_fit="contain", |
| height="auto" |
| ) |
|
|
| annotatedImage = gr.AnnotatedImage(color_map=COLOR_MAP) |
|
|
| @processButton.click(outputs=[processedImageGallery, annotatedImage, results], inputs=[inputImage, thresholdSlider, iouThresholdSlider]) |
| def Process(image, model_threshold, iouThresholdSlider): |
| if image is None: |
| raise gr.Warning("Lütfen görüntü yükleyiniz veya hazır seçiniz!", duration=3) |
| |
| img0, boxes, labels = DetectFaults(image, model_threshold, iouThresholdSlider) |
|
|
| if len(boxes) == 0: |
| raise gr.Error("Bir Hata ile Karşılaşıldı: Görüntüde Tespit Yapılamadı 💥!", duration=5) |
|
|
| sections = [] |
| for b, c in zip(boxes, labels): |
| sections+=[(b, LABELS[c])] |
|
|
| image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) |
| return [img0], (image, sections), "Görüntü İşlendi!" |
| |
| block.load(Warning) |
|
|
|
|
| block.queue(max_size=10) |
| block.launch(server_name="0.0.0.0", server_port=1071) |
|
|
|
|
