File size: 3,450 Bytes
b64d083 | 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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | import streamlit as st
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np
model=load_model('capstone_cnn_model.h5')
def process_image(img):
img=img.resize((32,32)) #boyutunu 32 x 32 pixel yaptik
img=np.array(img)
img=img/255.0 #normalize ettik
img=np.expand_dims(img,axis=0)
return img
st.title("Trafik Levhaları Tanıma :traffic_light: :vertical_traffic_light: :construction:")
st.write("Trafik levhası resmi ekleyin")
file=st.file_uploader('Bir Resim Sec',type=['jpg','jpeg','png'])
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)
class_names=["maksimum hız 20 km/h", #0
"maksimum hız 30 km/h", #1
"maksimum hız 50 km/h", #2
"maksimum hız 60 km/h", #3
"maksimum hız 70 km/h", #4
"maksimum hız 80 km/h" #5
"80 km/hhız limiti sonu",#6
"maksimum hız limiti 100 km/h", #7
"maksimum hız limiti 120 km/h", #8
"sollamak yasak", #9
"ağır tonajlı araçlara sollama yasağı", #10
"kavşak yaklaşıyor, öncelikli yoldasınız", #11
"geçiş öncelikli yol", #12
"yol ver", #13
"dur", #14
"motorlu araçlar giremez", #15
"ağır tonajlı araçlar için giriş yasağı", #16
"motorlu araçlar için giriş yasağı", #17
"dikkat", #18
"sola doğru viraj", #19
"sağa doğru viraj", #20
"soldan başlayan çift viraj", #21
"kasisli yol", #22
"kaygan yol", #23
"yol sağdan daralıyor", #24
"inşaat var", #25
"dikkat, trafik ışığı ile kontrol edilen kavşak yaklaşıyor", #26
"yayalara dikkat", #27
"dikkat, meskun mahal", #28
"bisikletlere dikkat", #29
"kar ve buzdan dolayı kayma tehlikesi", #30
"dikkat, yabani hayvan çıkabilir", #31
"tüm yasakların sonu", #32
"sağa dönme mecburiyeti", #33
"sola dönme mecburiyeti", #34
"düz devam etme mecburiyeti", #35
"düz gidilebilir veya sağa dönülebilir", #36
"sola dönülebilir veya düz gidilebilir", #37
"sağdan gidiniz", #38
"soldan gidiniz", #39
"dönel kavşak", #40
"sollama yasağı sonu", #41
"ağır tonajlı arabalara sollama yasağının sonu"] #42
st.write(class_names[predicted_class])
|