Initial commit.

.idea/workspace.xml

@@ -2,7 +2,6 @@
 <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" />
@@ -52,7 +51,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1647350746642</updated>
-      <workItem from="1647350750956" duration="11844000" />
+      <workItem from="1647350750956" duration="12425000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -369,7 +368,14 @@
       <option name="project" value="LOCAL" />
       <updated>1647362439904</updated>
     </task>
-    <option name="localTasksCounter" value="46" />
+    <task id="LOCAL-00046" summary="Initial commit.">
+      <created>1647363375249</created>
+      <option name="number" value="00046" />
+      <option name="presentableId" value="LOCAL-00046" />
+      <option name="project" value="LOCAL" />
+      <updated>1647363375249</updated>
+    </task>
+    <option name="localTasksCounter" value="47" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

app.py

@@ -14,6 +14,12 @@ from segmenter_model.utils import colorize_one, map2cs
 # WEIGHTS = './weights/segmenter.pth
 WEIGHTS = './weights/segmenter_nusc.pth'
 
+def blend_images(bg, fg, alpha=0.3):
+    fg = fg.convert('RGBA')
+    bg = bg.convert('RGBA')
+    blended = Image.blend(bg, fg, alpha=alpha)
+
+    return blended
 
 def download_file_from_google_drive(destination=WEIGHTS):
     id = '1v6_d2KHzRROsjb_cgxU7jvmnGVDXeBia'
@@ -132,9 +138,9 @@ model, window_size, window_stride = create_model()
 
 
 def predict(input_img):
-    input_img = Image.open(input_img)
+    input_img_pil = Image.open(input_img)
     transform = get_transformations()
-    input_img = transform(input_img)
+    input_img = transform(input_img_pil)
     input_img = torch.unsqueeze(input_img, 0)
 
     print('Loaded and prepaded image.')
@@ -152,8 +158,9 @@ def predict(input_img):
 
     # drawing_pseudo = transforms.ToPILImage()(drawing_pseudo)
     drawing_cs = transforms.ToPILImage()(drawing_cs)
+    drawing_cs_blend = blend_images(input_img_pil, drawing_cs)
     # return drawing_pseudo, drawing_cs
-    return drawing_cs
+    return drawing_cs_blend
 
 
 title = "Drive&Segment"