added files

README.md

@@ -1,8 +1,8 @@
 ---
 title: Traffic Segmentation
-emoji: 🏆
-colorFrom: indigo
-colorTo: yellow
+emoji: 🛣️🧠
+colorFrom: blue
+colorTo: red
 sdk: gradio
 sdk_version: 4.42.0
 app_file: app.py

app.py

@@ -0,0 +1,55 @@
+import gradio as gr
+from fastai.vision.all import *
+from PIL import Image, ImageDraw, ImageFont
+import matplotlib.pyplot as plt
+import numpy as np
+import matplotlib.colors as mcolors
+
+# import os
+# Load a pre-trained image classification model
+import pathlib
+plt = platform.system()
+if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
+if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
+
+root = os.path.dirname(__file__)
+
+
+model = load_learner("./models/model.pkl")
+
+def process(imagep):
+    # Predict and create the image as before
+    pred = model.predict(imagep)
+    a = pred[0]
+    i = np.stack([(a**17) % 255, (a**11) % 255, (a**9) % 255], axis=2)
+    img = Image.fromarray(i.astype('uint8'), mode='RGB')
+
+    imagep = Image.open(imagep)
+    # imagep = Image.fromarray(imagep)
+
+    imagep = imagep.convert("RGBA")
+    img = img.convert("RGBA")
+    img = img.resize(imagep.size)
+    alpha = Image.new('L', img.size, int(0.6 * 255))
+    img.putalpha(alpha)
+    combined = Image.alpha_composite(imagep, img)
+    
+    return combined.convert("RGB")
+
+
+
+# Sample images for user to choose from
+sample_images = ["./sample_images/street.jpg", "./sample_images/market.jpg","./sample_images/day.jpg"]
+
+iface = gr.Interface(
+    fn=process,
+    inputs=gr.Image(label="Select an image", type="filepath"),
+    outputs='image',
+    live=False,
+    title="Traffic image - semantic segmentation",
+    description="Upload a road traffic image or select one of the examples below",
+    examples=sample_images
+)
+
+
+iface.launch()

models/model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2901da8f72dfda0fbad5975cf28278ef44df1501d9c7c7b4488181028cfaba7f
+size 182564054

models/model_1.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f786ea24b7df31f4948a26f5617e124b678868fee63c5c7f9b4ae22a61075da5
+size 662857152

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+fastai==2.7.14
+spacy==3.7.4