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="8037000" />
+      <workItem from="1647350750956" duration="8171000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -220,7 +220,14 @@
       <option name="project" value="LOCAL" />
       <updated>1647358886682</updated>
     </task>
-    <option name="localTasksCounter" value="25" />
+    <task id="LOCAL-00025" summary="Initial commit.">
+      <created>1647358970013</created>
+      <option name="number" value="00025" />
+      <option name="presentableId" value="LOCAL-00025" />
+      <option name="project" value="LOCAL" />
+      <updated>1647358970013</updated>
+    </task>
+    <option name="localTasksCounter" value="26" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

app.py

@@ -134,12 +134,18 @@ def predict(input_img):
     input_img = transform(input_img)
     input_img = torch.unsqueeze(input_img, 0)
 
+    print('Loaded and prepaded image.')
+
     with torch.no_grad():
         segmentation = segment_segmenter(input_img, model, window_size, window_stride).squeeze().detach()
+        print('Segmented image.')
         segmentation_remap = remap(segmentation)
+        print('Remapped image.')
 
     drawing_pseudo = colorize_one(segmentation_remap)
+    print('Pseudo colors done.')
     drawing_cs = map2cs(segmentation_remap)
+    print('CS colors done.')
 
     # drawing_pseudo = transforms.ToPILImage()(drawing_pseudo)
     drawing_cs = transforms.ToPILImage()(drawing_cs)