Spaces:
Running
Running
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
|