Initial commit.

.idea/workspace.xml

@@ -2,6 +2,7 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="5dd22f22-8223-4d55-99f9-57d1e00622d7" name="Default Changelist" comment="Initial commit.">
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
@@ -50,7 +51,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1647350746642</updated>
-      <workItem from="1647350750956" duration="7582000" />
+      <workItem from="1647350750956" duration="7770000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -199,7 +200,14 @@
       <option name="project" value="LOCAL" />
       <updated>1647358382116</updated>
     </task>
-    <option name="localTasksCounter" value="22" />
+    <task id="LOCAL-00022" summary="Initial commit.">
+      <created>1647358512837</created>
+      <option name="number" value="00022" />
+      <option name="presentableId" value="LOCAL-00022" />
+      <option name="project" value="LOCAL" />
+      <updated>1647358512837</updated>
+    </task>
+    <option name="localTasksCounter" value="23" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

app.py

@@ -1,3 +1,5 @@
+import os
+
 import gradio as gr
 import numpy as np
 import requests
@@ -47,10 +49,10 @@ def download_file_from_google_drive(destination=WEIGHTS):
 
 
 def download_weights():
-    # if not os.path.exists(WEIGHTS):
-    url = 'https://data.ciirc.cvut.cz/public/projects/2022DriveAndSegment/segmenter.pth'
-    import urllib.request
-    urllib.request.urlretrieve(url, WEIGHTS)
+    if not os.path.exists(WEIGHTS):
+        url = 'https://data.ciirc.cvut.cz/public/projects/2022DriveAndSegment/segmenter.pth'
+        import urllib.request
+        urllib.request.urlretrieve(url, WEIGHTS)
 
 
 def segment_segmenter(image, model, window_size, window_stride, encoder_features=False, decoder_features=False,
@@ -139,9 +141,10 @@ def predict(input_img):
     drawing_pseudo = colorize_one(segmentation_remap)
     drawing_cs = map2cs(segmentation_remap)
 
-    drawing_pseudo = transforms.ToPILImage()(drawing_pseudo)
+    # drawing_pseudo = transforms.ToPILImage()(drawing_pseudo)
     drawing_cs = transforms.ToPILImage()(drawing_cs)
-    return drawing_pseudo, drawing_cs
+    # return drawing_pseudo, drawing_cs
+    return drawing_cs
 
 
 title = "Drive&Segment"