Spaces:
Sleeping
Sleeping
#!/usr/bin/env python | |
# coding: utf-8 | |
#dosyayı py olarak kaydet ve komut satırını kullanarak streamlit run streamlit.py | |
import streamlit as st | |
from tensorflow.keras.models import load_model | |
from PIL import Image | |
import numpy as np | |
import cv2 | |
model=load_model('GTS_cnn.h5') | |
def process_image(img): | |
img=img.resize((30,30)) | |
img=np.array(img) | |
img=img/255.0 | |
img=np.expand_dims(img,axis=0) | |
return img | |
st.title('GTS Trafik İşaretleri: :vertical_traffic_light: ') | |
st.write('Trafik işaretini yükleyin') | |
file=st.file_uploader('Bir resim seç', type= ['jpg','jpeg','png']) | |
class_names=[ | |
"Hız sınırı (20km/s)", "Hız sınırı (30km/s)", "Hız sınırı (50km/s)", "Hız sınırı (60km/s)", "Hız sınırı (70km/s)", | |
"Hız sınırı (80km/s)", "Hız sınırı sonu (80km/s)", "Hız sınırı (100km/s)", "Hız sınırı (120km/s)", "Geçiş yasağı", | |
"3.5 ton üstü araç geçişi yasak", "Kavşakta geçiş hakkı", "Ana yol", "Yol ver", "Dur", "Araç girişi yasak", | |
"3.5 ton üstü araç girişi yasak", "Giriş yasak", "Genel dikkat", "Tehlikeli sol viraj", "Tehlikeli sağ viraj", | |
"Çift viraj", "Engebeli yol", "Kaygan yol", "Yol sağdan daralıyor", "Yol çalışması", "Trafik ışıkları", "Yaya geçidi", | |
"Çocuk geçidi", "Bisiklet geçidi", "Buzlanma/kar dikkati", "Yabani hayvanlar geçebilir", "Hız ve geçiş limitlerinin sonu", | |
"İleri sağa dön", "İleri sola dön", "Sadece düz gidin", "Düz git veya sağa dön", "Düz git veya sola dön", "Sağdan devam et", | |
"Soldan devam et", "Döner kavşak zorunlu", "Geçiş yasağı sonu", "3.5 ton üstü geçiş yasağı sonu" | |
] | |
if file is not None: | |
img=Image.open(file) | |
st.image(img,caption='yuklenen resim') | |
image=process_image(img) | |
prediction=model.predict(image) | |
predicted_class=np.argmax(prediction) | |
st.write(class_names[predicted_class]) |