Initial commit.

segmenter_model/factory.py

@@ -49,7 +49,7 @@ def create_vit(model_cfg):
             # hard-coded for now, too lazy
             pretrained_weights = 'dino_deitsmall16_pretrain.pth'
             if not os.path.exists(pretrained_weights):
-                requests.get(pretrained_weights, allow_redirects=True)
+                requests.get('https://dl.fbaipublicfiles.com/dino/dino_deitsmall16_pretrain/dino_deitsmall16_pretrain.pth', allow_redirects=True)
             model.load_state_dict(torch.load(pretrained_weights), strict=True)
         else:
             model = torch.hub.load('facebookresearch/dino:main', backbone)