From 08d3536e25af254132fb06b679baea7daec2689a Mon Sep 17 00:00:00 2001
From: ptresson <paul.tresson@ird.fr>
Date: Tue, 8 Oct 2024 16:34:56 +0200
Subject: [PATCH] sam back as default since its no issue with high enough
versions of timm
---
encoder.py | 4 ++--
tests/test_encoder.py | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/encoder.py b/encoder.py
index d9dccb1..1b77a66 100644
--- a/encoder.py
+++ b/encoder.py
@@ -248,14 +248,14 @@ class EncoderAlgorithm(QgsProcessingAlgorithm):
'ViT base DINO',
'ViT tiny Imagenet (smallest)',
'ViT base MAE',
- # 'SAM',
+ 'SAM',
'--Empty--'
]
self.timm_backbone_opt = [
'vit_base_patch16_224.dino',
'vit_tiny_patch16_224.augreg_in21k',
'vit_base_patch16_224.mae',
- # 'samvit_base_patch16.sa1b',
+ 'samvit_base_patch16.sa1b',
]
self.addParameter (
QgsProcessingParameterEnum(
diff --git a/tests/test_encoder.py b/tests/test_encoder.py
index 948c453..cc3dd41 100644
--- a/tests/test_encoder.py
+++ b/tests/test_encoder.py
@@ -98,13 +98,13 @@ class TestEncoderAlgorithm(unittest.TestCase):
'vit_base_patch16_224.dino',
'vit_tiny_patch16_224.augreg_in21k',
'vit_base_patch16_224.mae',
- # 'samvit_base_patch16.sa1b',
+ 'samvit_base_patch16.sa1b',
]
expected_output_size = [
torch.Size([1,197,768]),
torch.Size([1,197,192]),
torch.Size([1,197,768]),
- # torch.Size([1, 256, 64, 64]),
+ torch.Size([1, 256, 64, 64]),
]
for arch, exp_feat_size in zip(archs, expected_output_size):
--
GitLab