File size: 664 Bytes
612a2c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import torch
import cv2
import numpy as np
import gradio as gr
from ultralytics import YOLO
from PIL import Image
from huggingface_hub import hf_hub_download
# Hugging Face'ten YOLO modelini indir
model_path = hf_hub_download("keremberke/yolov8n-pcb-defect-segmentation", "best.pt")
# Modeli yükle
model = YOLO(model_path)
def detect_defects(image):
results = model.predict(image, save=False)
annotated_image = results[0].plot() # Tahmin sonuçlarını çiz
return annotated_image
iface = gr.Interface(
fn=detect_defects,
inputs=gr.Image(type="pil"),
outputs=gr.Image(type="pil"),
title="PCB Defect Segmentation"
)
iface.launch() |