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

fix typos and change default model to dino small patch 8

parent 07a7d7ad
No related branches found
No related tags found
No related merge requests found
...@@ -172,7 +172,7 @@ class EncoderAlgorithm(IAMAPAlgorithm): ...@@ -172,7 +172,7 @@ class EncoderAlgorithm(IAMAPAlgorithm):
tmp_files_cleanup_frq = QgsProcessingParameterNumber( tmp_files_cleanup_frq = QgsProcessingParameterNumber(
name=self.TEMP_FILES_CLEANUP_FREQ, name=self.TEMP_FILES_CLEANUP_FREQ,
description=self.tr( description=self.tr(
"Frequencie at which temporary files should be cleaned up (zero means no cleanup)." "Frequency at which temporary files should be cleaned up (zero means no cleanup)."
), ),
type=QgsProcessingParameterNumber.Integer, type=QgsProcessingParameterNumber.Integer,
defaultValue=500, defaultValue=500,
...@@ -247,6 +247,7 @@ class EncoderAlgorithm(IAMAPAlgorithm): ...@@ -247,6 +247,7 @@ class EncoderAlgorithm(IAMAPAlgorithm):
) )
) )
self.backbone_opt = [ self.backbone_opt = [
"ViT small DINO patch 8",
"ViT base DINO", "ViT base DINO",
"ViT tiny Imagenet (smallest)", "ViT tiny Imagenet (smallest)",
"ViT base MAE", "ViT base MAE",
...@@ -254,6 +255,7 @@ class EncoderAlgorithm(IAMAPAlgorithm): ...@@ -254,6 +255,7 @@ class EncoderAlgorithm(IAMAPAlgorithm):
"--Empty--", "--Empty--",
] ]
self.timm_backbone_opt = [ self.timm_backbone_opt = [
"vit_samll_patch8_224.dino",
"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",
...@@ -631,7 +633,7 @@ class EncoderAlgorithm(IAMAPAlgorithm): ...@@ -631,7 +633,7 @@ class EncoderAlgorithm(IAMAPAlgorithm):
# overwritting merged_tmp.tif may be impossible in windows (e.g. if an antivirus is analysing the newly created data) # overwritting merged_tmp.tif may be impossible in windows (e.g. if an antivirus is analysing the newly created data)
# then, merging and cleaning is impossible # then, merging and cleaning is impossible
except Exception as e : except Exception as e :
feedback.pushWarning(f"Unable to cleaning temporary files ! Try to delete them latter at {self.output_subdir}") feedback.pushWarning(f"Unable to clean temporary files ! Try to delete them latter at {self.output_subdir}")
self.all_encoding_done = True self.all_encoding_done = True
......
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