feat: decrease margin choices

vra · Jul 1, 2023 · 4b8e2e9 · 4b8e2e9
1 parent 714c481
commit 4b8e2e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dinov2_retrieval/__init__.py b/src/dinov2_retrieval/__init__.py
@@ -59,7 +59,7 @@ def main():
         "--margin",
         type=int,
         default=10,
-        choices=range(0, 100),
+        choices=range(0, 105, 5),
         help="margin size (in pixel) between concatenated images",
     )
     parser.add_argument(