demo added

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 Oğuzcan Turan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,13 +1,11 @@
 ---
-title: Satellighte
-emoji: 📈
-colorFrom: pink
-colorTo: pink
+title: Satellighte Streamlit
+emoji: 📡
+colorFrom: blue
+colorTo: purple
 sdk: streamlit
 sdk_version: 1.10.0
 app_file: app.py
 pinned: false
 license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---

app.py

@@ -0,0 +1,92 @@
+import random
+from datetime import datetime
+
+import numpy as np
+import requests
+import satellighte as sat
+import streamlit as st
+from PIL import Image
+
+
+def main():
+ # pylint: disable=no-member
+
+ st.set_page_config(
+ page_title="Satellighte Demo Page",
+ page_icon="📡",
+ layout="centered",
+ initial_sidebar_state="expanded",
+ menu_items={
+ "Get Help": "https://canturan10.github.io/satellighte/",
+ "About": "Satellite Image Classification",
+ },
+ )
+
+ st.title("Satellighte Demo Page")
+
+ url = "https://raw.githubusercontent.com/canturan10/satellighte/master/src/satellighte.png?raw=true"
+ satellighte = Image.open(requests.get(url, stream=True).raw)
+
+ st.sidebar.image(satellighte, width=100)
+ st.sidebar.title("Satellighte")
+ st.sidebar.caption(sat.__description__)
+
+ st.write(
+ "**Satellighte** is an image classification library that consist state-of-the-art deep learning methods. It is a combination of the words **'Satellite'** and **'Light'**, and its purpose is to establish a light structure to classify satellite images, but to obtain robust results."
+ )
+
+ st.sidebar.caption(f"Version: `{sat.__version__}`")
+ st.sidebar.caption(f"License: `{sat.__license__}`")
+ st.sidebar.caption(sat.__copyright__)
+
+ selected_model = st.selectbox(
+ "Select model",
+ sat.available_models(),
+ )
+ selected_version = st.selectbox(
+ "Select version",
+ sat.get_model_versions(selected_model),
+ )
+
+ model = sat.Classifier.from_pretrained(selected_model, selected_version)
+ model.eval()
+
+ uploaded_file = st.file_uploader(
+ "", type=["png", "jpg", "jpeg"], accept_multiple_files=False
+ )
+
+ if uploaded_file is None:
+ st.write("Sample Image")
+ # Sample image.
+ url = f"https://raw.githubusercontent.com/canturan10/satellighte/master/src/eurosat_samples/{random_sample}?raw=true"
+ image = Image.open(requests.get(url, stream=True).raw)
+
+ else:
+ # User-selected image.
+ image = Image.open(uploaded_file)
+
+ image = np.array(image.convert("RGB"))
+ FRAME_WINDOW = st.image([], use_column_width=True)
+
+ model = sat.Classifier.from_pretrained(selected_model, selected_version)
+ model.eval()
+
+ results = model.predict(image)
+ pil_img = sat.utils.visualize(image, results)
+
+ st.write("Results:", results)
+ FRAME_WINDOW.image(pil_img)
+
+
+if __name__ == "__main__":
+ samples = [
+ "AnnualCrop.jpg",
+ "Forest.jpg",
+ "HerbaceousVegetation.jpg",
+ "PermanentCrop.jpg",
+ "River.jpg",
+ ]
+ random.seed(datetime.now())
+ random_sample = samples[random.randint(0, len(samples) - 1)]
+
+ main()

requirements.txt

@@ -0,0 +1,2 @@
+satellighte
+streamlit~=1.11.1