mudassir032 commited on
Commit
80058cc
·
verified ·
1 Parent(s): 7e82b03

Upload 6 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ Chest[[:space:]]Xray/pneumonia.keras filter=lfs diff=lfs merge=lfs -text
Chest Xray/chest-xray.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import numpy as np
2
+ import cv2
3
+ import tensorflow as tf
4
+ import streamlit as st
5
+ from tensorflow.keras.models import load_model
6
+
7
+ st.title("Pnuemonia")
8
+ def load_img(path):
9
+ img = cv2.imread(path)
10
+ img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
11
+ img = cv2.resize(img,(224,224))
12
+ img = img.astype('float32')
13
+ img /= 255.0
14
+ return img
15
+
16
+ imgs = ['person1676_virus_2892.jpeg','1-s2.0-S0263931909001811-gr3.jpg','43055_2020_296_Fig11_HTML.png','images.jpg']
17
+ @st.cache_resource
18
+ def load_cached_models():
19
+ model = load_model("pneumonia_vgg16.keras")
20
+ model1 = load_model("pnuemoniavgg16_1.keras")
21
+ model2 = load_model("pnuemoniavgg16_2.keras")
22
+ return model, model1, model2
23
+
24
+ # Load the models once, cached in memory
25
+ model, model1, model2 = load_cached_models()
26
+ for img_p in imgs:
27
+ img = load_img(img_p)
28
+ img = np.expand_dims(img,axis=0)
29
+ pred = model.predict(img)
30
+ pred1 = model1.predict(img)
31
+ pred2 = model2.predict(img)
32
+ avg_pred = np.mean([pred, pred1, pred2])
33
+ st.subheader(img_p)
34
+ st.write("Prediction of model")
35
+ final_pred = round(avg_pred*100,2)
36
+ st.progress(int(round(avg_pred * 100)))
37
+ st.write(final_pred)
38
+ st.write("Pneumonia Detected" if avg_pred > .5 else "Your x-ray seems normal")
Chest Xray/pneumonia.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9555e9a3b0a40b5caba287d24c560076bb8c1a792d4bf01702b38ab79179793
3
+ size 136022360
Chest Xray/test_imgs/1-s2.0-S0263931909001811-gr3.jpg ADDED
Chest Xray/test_imgs/43055_2020_296_Fig11_HTML.png ADDED
Chest Xray/test_imgs/images.jpg ADDED
Chest Xray/test_imgs/person1676_virus_2892.jpeg ADDED