Duplicate from adpro/semantic-03

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.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>

README.md

@@ -0,0 +1,14 @@
+---
+title: Semantic Segmentation Segformer
+emoji: 📊
+colorFrom: indigo
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.3
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: adpro/semantic-03
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,36 @@
+import gluoncv
+import mxnet as mx
+from gluoncv.utils.viz import get_color_pallete
+import gradio as gr
+import numpy as np
+from PIL import Image
+from gluoncv.data.transforms.presets.segmentation import test_transform
+
+# using cpu
+ctx = mx.cpu(0)
+
+model = gluoncv.model_zoo.get_model("psp_resnet101_ade", pretrained=True)
+
+
+def segmentation(image):
+    img = Image.fromarray(image)
+    img = mx.ndarray.array(img)
+    img = test_transform(img, ctx)
+    output = model.predict(img)
+    predict = mx.nd.squeeze(mx.nd.argmax(output, 1)).asnumpy()
+    mask = get_color_pallete(predict, "ade20k")
+    return mask
+
+
+image_in = gr.Image()
+image_out = gr.components.Image()
+description = "MXNet Image Segmentation Model"
+examples=['cat.jpeg']
+
+Iface = gr.Interface(
+    fn=segmentation,
+    inputs=image_in,
+    outputs=image_out,
+    title="Semantic Segmentation - MXNet",
+    examples=examples
+).launch()

requirements.txt

@@ -0,0 +1,7 @@
+mxnet==1.9.1
+gluoncv==0.10.5
+matplotlib>=3.5.3
+opencv-python>=4.6.0
+gradio
+numpy==1.23.1
+pillow