From 4b8e2e901b5bf43293ef54c822d8d8b09bf1e2c9 Mon Sep 17 00:00:00 2001
From: weishan <weishan.wyf@alibaba-inc.com>
Date: Sat, 1 Jul 2023 22:47:52 +0800
Subject: [PATCH] feat: decrease margin choices

---
 src/dinov2_retrieval/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dinov2_retrieval/__init__.py b/src/dinov2_retrieval/__init__.py
index 2fed2c7..2fa16ad 100644
--- 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(