|
from ultralytics import YOLO |
|
import cv2 |
|
from func import plate_reader |
|
from sort.sort import * |
|
import os |
|
from PIL import Image |
|
from util import get_car |
|
import streamlit as st |
|
from micro_ocr import plate_rec |
|
|
|
yolo_model = YOLO('yolov8n.pt') |
|
license_plate_detector = YOLO('models/plate.pt') |
|
|
|
|
|
uploaded_file = st.file_uploader("Choose a file") |
|
if uploaded_file is not None: |
|
image = Image.open(uploaded_file) |
|
|
|
st.image(image) |
|
img_np = np.array(image) |
|
if os.path.exists("img.jpg"): |
|
os.remove("img.jpg") |
|
cv2.imwrite("img.jpg",img_np) |
|
|
|
|
|
|
|
if st.button("Bu bir düğme"): |
|
|
|
text, p = plate_reader(license_plate_detector,yolo_model,"img.jpg") |
|
|
|
|
|
|
|
st.image(image) |
|
st.image(p) |
|
st.text(text) |
|
|
|
|
|
|
|
|