Skip to content
Snippets Groups Projects
Commit 08d3536e authored by paul.tresson_ird.fr's avatar paul.tresson_ird.fr
Browse files

sam back as default since its no issue with high enough versions of timm

parent 3c050406
No related branches found
No related tags found
No related merge requests found
...@@ -248,14 +248,14 @@ class EncoderAlgorithm(QgsProcessingAlgorithm): ...@@ -248,14 +248,14 @@ class EncoderAlgorithm(QgsProcessingAlgorithm):
'ViT base DINO', 'ViT base DINO',
'ViT tiny Imagenet (smallest)', 'ViT tiny Imagenet (smallest)',
'ViT base MAE', 'ViT base MAE',
# 'SAM', 'SAM',
'--Empty--' '--Empty--'
] ]
self.timm_backbone_opt = [ self.timm_backbone_opt = [
'vit_base_patch16_224.dino', 'vit_base_patch16_224.dino',
'vit_tiny_patch16_224.augreg_in21k', 'vit_tiny_patch16_224.augreg_in21k',
'vit_base_patch16_224.mae', 'vit_base_patch16_224.mae',
# 'samvit_base_patch16.sa1b', 'samvit_base_patch16.sa1b',
] ]
self.addParameter ( self.addParameter (
QgsProcessingParameterEnum( QgsProcessingParameterEnum(
......
...@@ -98,13 +98,13 @@ class TestEncoderAlgorithm(unittest.TestCase): ...@@ -98,13 +98,13 @@ class TestEncoderAlgorithm(unittest.TestCase):
'vit_base_patch16_224.dino', 'vit_base_patch16_224.dino',
'vit_tiny_patch16_224.augreg_in21k', 'vit_tiny_patch16_224.augreg_in21k',
'vit_base_patch16_224.mae', 'vit_base_patch16_224.mae',
# 'samvit_base_patch16.sa1b', 'samvit_base_patch16.sa1b',
] ]
expected_output_size = [ expected_output_size = [
torch.Size([1,197,768]), torch.Size([1,197,768]),
torch.Size([1,197,192]), torch.Size([1,197,192]),
torch.Size([1,197,768]), 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): for arch, exp_feat_size in zip(archs, expected_output_size):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment