Spaces:
Running
Running
| import streamlit as st | |
| import keras | |
| import numpy as np | |
| from PIL import Image | |
| import io | |
| import urllib.request | |
| import os | |
| st.set_page_config(layout="wide") | |
| st.markdown(""" | |
| <style> | |
| .block-container { | |
| padding-top: 1rem; | |
| padding-bottom: 0rem; | |
| padding-left: 5rem; | |
| padding-right: 5rem; | |
| } | |
| </style> | |
| """, unsafe_allow_html=True) | |
| #title | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| buffer, col = st.columns(2) | |
| with col: | |
| st.header('Overpass Identifier') | |
| with col2: | |
| buffer1, col, buffer2 = st.columns(3) | |
| with col: | |
| st.image('overpass.png') | |
| st.write("---") | |
| #load model and initialize image size required by model. uploaded images are resized to indicated size | |
| img_height = 640 | |
| img_width = 640 | |
| state = st.session_state | |
| if "loaded_model" not in state: | |
| with st.spinner('Loading model. This may take a few seconds...'): | |
| state.loaded_model = keras.models.load_model("0.0008-0.92.keras") | |
| if "lat" not in state: | |
| state.lat = 39.11 | |
| if "lng" not in state: | |
| state.lng = -86.56 | |
| if "coords_submitted" not in state: | |
| state.coords_submitted = False | |
| #if "img_submitted" not in state: | |
| #state.img_submitted = False | |
| if "img" not in state: | |
| state.img = None | |
| col1, col2, col3 = st.columns(3) | |
| with col3: | |
| #header | |
| st.subheader('Enter latitude/longitude coordinates:') | |
| coll, colr= st.columns(2) | |
| with coll: | |
| state.lat = st.number_input('Latitude', value=39.11, min_value=-90., max_value=90., step=.01) | |
| st.write('The current lat/long are:') | |
| with colr: | |
| state.lng = st.number_input('Longitude', value=-86.56, min_value=-180., max_value=180., step=.01) | |
| st.write(str(state.lat)+', '+str(state.lng)) | |
| with st.form("my_form"): | |
| submit_button = st.form_submit_button( | |
| label="Get Image and Prediction", on_click=lambda: state.update(coords_submitted=True)) | |
| #header | |
| #st.subheader('Upload a satellite image:') | |
| #img_buffer = st.file_uploader("Upload a satellite image file (format: .png, .jpeg, or .jpg).",type=['png', 'jpeg', 'jpg']) | |
| #if img_buffer is not None: | |
| #state.img = Image.open(img_buffer).convert("RGB") | |
| with col2: | |
| if state.coords_submitted: | |
| state.coords_submitted = False | |
| try: | |
| api_key = os.getenv("goog_api") | |
| url = "https://maps.googleapis.com/maps/api/staticmap?center="+str(state.lat)+","+str(state.lng)+"&zoom=16&size=640x640&maptype=satellite&key="+api_key | |
| buffer = io.BytesIO(urllib.request.urlopen(url).read()) | |
| state.img = Image.open(buffer).convert("RGB") | |
| except Exception as e: | |
| st.write("Error! Could not access Google Static API. Error code:",e) | |
| if state.img is not None: | |
| st.image(state.img, use_column_width = True) | |
| with col1: | |
| st.subheader("Prediction") | |
| if state.img is not None: | |
| img_array = np.array(state.img) | |
| batch_size = 1 | |
| img_array = np.reshape(img_array,[batch_size,img_height,img_width,3]) | |
| result = state.loaded_model.predict(img_array) | |
| crossing_chance = result[0][1]*100 | |
| status = None | |
| while status is None: | |
| if crossing_chance >= 0: | |
| status = "extremely un" | |
| if crossing_chance >= 10: | |
| status = "highly un" | |
| if crossing_chance >= 20: | |
| status = "pretty un" | |
| if crossing_chance >= 30: | |
| status = "slightly un" | |
| if crossing_chance >= 40: | |
| status = "a tiny bit more unlikely than " | |
| if crossing_chance >= 50: | |
| status = "a tiny bit more likely than un" | |
| if crossing_chance >= 60: | |
| status = "slightly " | |
| if crossing_chance >= 70: | |
| status = "pretty " | |
| if crossing_chance >= 80: | |
| status = "highly " | |
| if crossing_chance >= 90: | |
| status = "extremely " | |
| st.write(f"It's {status}likely there's at least one overpass here.") | |
| st.write("") | |
| st.write(f"In fact, the likelihood of at least one overpass is {np.round(crossing_chance,decimals=2)}%.") | |