Initial commit.

.idea/workspace.xml

@@ -1,9 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <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$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
-    </list>
+    <list default="true" id="5dd22f22-8223-4d55-99f9-57d1e00622d7" name="Default Changelist" comment="Initial commit." />
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
     <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
@@ -50,7 +48,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1647350746642</updated>
-      <workItem from="1647350750956" duration="4786000" />
+      <workItem from="1647350750956" duration="5088000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -101,7 +99,28 @@
       <option name="project" value="LOCAL" />
       <updated>1647355534027</updated>
     </task>
-    <option name="localTasksCounter" value="8" />
+    <task id="LOCAL-00008" summary="Initial commit.">
+      <created>1647355606191</created>
+      <option name="number" value="00008" />
+      <option name="presentableId" value="LOCAL-00008" />
+      <option name="project" value="LOCAL" />
+      <updated>1647355606191</updated>
+    </task>
+    <task id="LOCAL-00009" summary="Initial commit.">
+      <created>1647355689933</created>
+      <option name="number" value="00009" />
+      <option name="presentableId" value="LOCAL-00009" />
+      <option name="project" value="LOCAL" />
+      <updated>1647355689933</updated>
+    </task>
+    <task id="LOCAL-00010" summary="Initial commit.">
+      <created>1647355817619</created>
+      <option name="number" value="00010" />
+      <option name="presentableId" value="LOCAL-00010" />
+      <option name="project" value="LOCAL" />
+      <updated>1647355817619</updated>
+    </task>
+    <option name="localTasksCounter" value="11" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

requirements.txt

@@ -3,5 +3,4 @@ torchvision
 pillow
 timm
 pyyaml
-einops
-opencv-python
+einops

segmenter_model/utils.py

@@ -3,11 +3,9 @@ import math
 # from segm.engine import seg2rgb
 from collections import namedtuple
 
-import cv2
 import numpy as np
 import torch.nn as nn
 import torch.nn.functional as F
-from PIL import Image
 from timm.models.layers import trunc_normal_
 
 import torch
@@ -333,18 +331,6 @@ def inference_picie(
                 seg_maps[i: i + WB] = probs
         windows["seg_maps"] = seg_maps
 
-        if debug_file is not None:
-            if isinstance(im_rgb, torch.Tensor):
-                im_rgb = im_rgb.detach().cpu().numpy()
-            if len(im_rgb.shape) == 4:
-                im_rgb = im_rgb[0]
-            h, w = im.shape[-2:]
-            im_rgb = cv2.resize(im_rgb, (w, h), interpolation=cv2.INTER_LINEAR)
-
-            crops_rgb = np.stack(
-                sliding_window(im_rgb[None, :], flip, window_size, window_stride, channels_first=channel_first).pop(
-                    "crop"))[:, 0]
-
         im_seg_map = merge_windows(windows, window_size, ori_shape, no_softmax=decoder_features,
                                    no_upsample=no_upsample, patch_size=None)
 
@@ -412,41 +398,6 @@ def inference(
                 # torch.cuda.empty_cache()
         windows["seg_maps"] = seg_maps
 
-        if debug_file is not None:
-            if isinstance(im_rgb, torch.Tensor):
-                im_rgb = im_rgb.detach().cpu().numpy()
-            if len(im_rgb.shape) == 4:
-                im_rgb = im_rgb[0]
-            h, w = im.shape[-2:]
-            im_rgb = cv2.resize(im_rgb, (w, h), interpolation=cv2.INTER_LINEAR)
-
-            crops_rgb = np.stack(
-                sliding_window(im_rgb[None, :], flip, window_size, window_stride, channels_first=channel_first).pop(
-                    "crop"))[:, 0]
-
-            windows_row = np.concatenate([w for w in crops_rgb], axis=1)
-            # print(windows_row)
-            try:
-                Image.fromarray(windows_row).save(debug_file)
-            except:
-                pass
-
-            suffix = debug_file[-4:]
-            debug_file = debug_file.replace(suffix, '_preds{}'.format(suffix))
-            windows_preds = seg_maps.argmax(dim=1).cpu().numpy()
-            windows_preds_row = np.concatenate([seg2rgb(wp, C, 255) for wp in windows_preds], axis=1)
-            windows_row_plus_preds = np.concatenate((windows_row, windows_preds_row), axis=0)
-            try:
-                Image.fromarray(windows_preds_row).save(debug_file)
-            except:
-                pass
-
-            debug_file = debug_file.replace(suffix, '_wImg{}'.format(suffix))
-            try:
-                Image.fromarray(windows_row_plus_preds).save(debug_file)
-            except:
-                pass
-
         im_seg_map = merge_windows(windows, window_size, ori_shape, no_softmax=decoder_features,
                                    no_upsample=no_upsample, patch_size=model.patch_size)