Initial commit.

.idea/workspace.xml

@@ -50,7 +50,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1647350746642</updated>
-      <workItem from="1647350750956" duration="10459000" />
+      <workItem from="1647350750956" duration="10672000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -325,7 +325,14 @@
       <option name="project" value="LOCAL" />
       <updated>1647361256773</updated>
     </task>
-    <option name="localTasksCounter" value="40" />
+    <task id="LOCAL-00040" summary="Initial commit.">
+      <created>1647361647000</created>
+      <option name="number" value="00040" />
+      <option name="presentableId" value="LOCAL-00040" />
+      <option name="project" value="LOCAL" />
+      <updated>1647361647000</updated>
+    </task>
+    <option name="localTasksCounter" value="41" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

app.py

@@ -50,10 +50,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_nusc.pth'
+    import urllib.request
+    urllib.request.urlretrieve(url, WEIGHTS)
 
 
 def segment_segmenter(image, model, window_size, window_stride, encoder_features=False, decoder_features=False,