added

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,66 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="53">
+            <item index="0" class="java.lang.String" itemvalue="pathspec" />
+            <item index="1" class="java.lang.String" itemvalue="transformers" />
+            <item index="2" class="java.lang.String" itemvalue="huggingface-hub" />
+            <item index="3" class="java.lang.String" itemvalue="numpy" />
+            <item index="4" class="java.lang.String" itemvalue="importlib-metadata" />
+            <item index="5" class="java.lang.String" itemvalue="exceptiongroup" />
+            <item index="6" class="java.lang.String" itemvalue="fastapi" />
+            <item index="7" class="java.lang.String" itemvalue="filelock" />
+            <item index="8" class="java.lang.String" itemvalue="platformdirs" />
+            <item index="9" class="java.lang.String" itemvalue="zipp" />
+            <item index="10" class="java.lang.String" itemvalue="starlette" />
+            <item index="11" class="java.lang.String" itemvalue="urllib3" />
+            <item index="12" class="java.lang.String" itemvalue="uvicorn" />
+            <item index="13" class="java.lang.String" itemvalue="joblib" />
+            <item index="14" class="java.lang.String" itemvalue="mypy" />
+            <item index="15" class="java.lang.String" itemvalue="nltk" />
+            <item index="16" class="java.lang.String" itemvalue="python-dateutil" />
+            <item index="17" class="java.lang.String" itemvalue="mypy-extensions" />
+            <item index="18" class="java.lang.String" itemvalue="pycodestyle" />
+            <item index="19" class="java.lang.String" itemvalue="setuptools" />
+            <item index="20" class="java.lang.String" itemvalue="cycler" />
+            <item index="21" class="java.lang.String" itemvalue="flake8" />
+            <item index="22" class="java.lang.String" itemvalue="markupsafe" />
+            <item index="23" class="java.lang.String" itemvalue="mccabe" />
+            <item index="24" class="java.lang.String" itemvalue="tomli" />
+            <item index="25" class="java.lang.String" itemvalue="itsdangerous" />
+            <item index="26" class="java.lang.String" itemvalue="pyparsing" />
+            <item index="27" class="java.lang.String" itemvalue="werkzeug" />
+            <item index="28" class="java.lang.String" itemvalue="six" />
+            <item index="29" class="java.lang.String" itemvalue="pyflakes" />
+            <item index="30" class="java.lang.String" itemvalue="black" />
+            <item index="31" class="java.lang.String" itemvalue="kiwisolver" />
+            <item index="32" class="java.lang.String" itemvalue="setuptools-scm" />
+            <item index="33" class="java.lang.String" itemvalue="wordcloud" />
+            <item index="34" class="java.lang.String" itemvalue="packaging" />
+            <item index="35" class="java.lang.String" itemvalue="typing-extensions" />
+            <item index="36" class="java.lang.String" itemvalue="click" />
+            <item index="37" class="java.lang.String" itemvalue="contourpy" />
+            <item index="38" class="java.lang.String" itemvalue="jinja2" />
+            <item index="39" class="java.lang.String" itemvalue="pandas" />
+            <item index="40" class="java.lang.String" itemvalue="tqdm" />
+            <item index="41" class="java.lang.String" itemvalue="fonttools" />
+            <item index="42" class="java.lang.String" itemvalue="regex" />
+            <item index="43" class="java.lang.String" itemvalue="matplotlib" />
+            <item index="44" class="java.lang.String" itemvalue="pillow" />
+            <item index="45" class="java.lang.String" itemvalue="isort" />
+            <item index="46" class="java.lang.String" itemvalue="pytz" />
+            <item index="47" class="java.lang.String" itemvalue="flask" />
+            <item index="48" class="java.lang.String" itemvalue="textblob" />
+            <item index="49" class="java.lang.String" itemvalue="nemo-toolkit" />
+            <item index="50" class="java.lang.String" itemvalue="gradio" />
+            <item index="51" class="java.lang.String" itemvalue="pytube" />
+            <item index="52" class="java.lang.String" itemvalue="nemo" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (semantic-segmentation-segformer)" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/semantic-segmentation-segformer.iml" filepath="$PROJECT_DIR$/.idea/semantic-segmentation-segformer.iml" />
+    </modules>
+  </component>
+</project>

.idea/semantic-segmentation-segformer.iml

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="Python 3.9 (semantic-segmentation-segformer)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PackageRequirementsSettings">
+    <option name="versionSpecifier" value="Greater or equal (&gt;=x.y.z)" />
+  </component>
+</module>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -0,0 +1,37 @@
+import gluoncv
+import matplotlib.image as mpimg
+import mxnet as mx
+from gluoncv.data.transforms.presets.segmentation import test_transform
+from gluoncv.utils.viz import get_color_pallete
+from matplotlib import pyplot as plt
+from mxnet import image
+import gradio as gr
+import numpy as np
+
+# using cpu
+ctx = mx.cpu(0)
+
+
+FILE_NAME = "result.png"
+
+model = gluoncv.model_zoo.get_model("psp_resnet101_ade", pretrained=True)
+
+def segmentation(img):
+    output = model.predict(img)
+    predict = mx.nd.squeeze(mx.nd.argmax(output, 1)).asnumpy()
+    mask = get_color_pallete(predict, "ade20k")
+    # mask.save("result.png")
+    # mmask = mpimg.imread("result.png")
+    # plt.imshow(mmask)
+    # plt.savefig("result.png")
+    return mask
+
+image_in = gr.Image()
+image_out = gr.components.Image()
+
+Iface = gr.Interface(
+    fn=segmentation,
+    inputs=image_in,
+    outputs=image_out,
+    title="Sementic Segmentation - MXNet",
+).launch()

requirements.txt

@@ -0,0 +1,5 @@
+mxnet>=1.9.1
+gluoncv>=0.10.5
+matplotlib>=3.5.3
+opencv-python>=4.6.0
+gradio