HsbYldz commited on
Commit
b64d083
1 Parent(s): a60476d

Upload 3 files

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