mertbozkurt's picture
Update app.py
3437949
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)
#image = image.rotate(270, expand=True)
st.image(image)
img_np = np.array(image)
if os.path.exists("img.jpg"):
os.remove("img.jpg")
cv2.imwrite("img.jpg",img_np)
# Basit bir düğme ekleyin
if st.button("Bu bir düğme"):
#img = cv2.imread("img.jpg")
text, p = plate_reader(license_plate_detector,yolo_model,"img.jpg")
st.image(image)
st.image(p)
st.text(text)