Initial commit.

.idea/workspace.xml

@@ -2,8 +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" />
+      <change beforePath="$PROJECT_DIR$/segmenter_model/factory.py" beforeDir="false" afterPath="$PROJECT_DIR$/segmenter_model/factory.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -51,7 +50,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1647350746642</updated>
-      <workItem from="1647350750956" duration="5496000" />
+      <workItem from="1647350750956" duration="5731000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -137,7 +136,14 @@
       <option name="project" value="LOCAL" />
       <updated>1647356274640</updated>
     </task>
-    <option name="localTasksCounter" value="13" />
+    <task id="LOCAL-00013" summary="Initial commit.">
+      <created>1647356326582</created>
+      <option name="number" value="00013" />
+      <option name="presentableId" value="LOCAL-00013" />
+      <option name="project" value="LOCAL" />
+      <updated>1647356326582</updated>
+    </task>
+    <option name="localTasksCounter" value="14" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

segmenter_model/factory.py

@@ -1,18 +1,17 @@
-from pathlib import Path
-import yaml
-import torch
-import math
 import os
-import torch.nn as nn
+from pathlib import Path
 
+import requests
+import yaml
 from timm.models.helpers import load_pretrained, load_custom_pretrained
-from timm.models.vision_transformer import default_cfgs, checkpoint_filter_fn
 from timm.models.registry import register_model
 from timm.models.vision_transformer import _create_vision_transformer
+from timm.models.vision_transformer import default_cfgs, checkpoint_filter_fn
+
+import segmenter_model.torch as ptu
+import torch
 from segmenter_model.decoder import MaskTransformer
 from segmenter_model.segmenter import Segmenter
-import segmenter_model.torch as ptu
-
 from segmenter_model.vit_dino import vit_small, VisionTransformer
 
 
@@ -48,14 +47,9 @@ def create_vit(model_cfg):
             model_cfg['drop_rate'] = model_cfg['dropout']
             model = vit_small(**model_cfg)
             # hard-coded for now, too lazy
-            ciirc_path = '/home/vobecant/PhD/weights/dino/dino_deitsmall16_pretrain.pth'
-            karolina_path = '/scratch/project/dd-21-20/pretrained_weights/dino/dino_deitsmall16_pretrain.pth'
-            if os.path.exists(ciirc_path):
-                pretrained_weights = ciirc_path
-            elif os.path.exists(karolina_path):
-                pretrained_weights = karolina_path
-            else:
-                raise Exception('DINO weights not found!')
+            pretrained_weights = 'dino_deitsmall16_pretrain.pth'
+            if not os.path.exists(pretrained_weights):
+                requests.get(pretrained_weights, allow_redirects=True)
             model.load_state_dict(torch.load(pretrained_weights), strict=True)
         else:
             model = torch.hub.load('facebookresearch/dino:main', backbone)