diff --git a/sdcat/cluster/utils.py b/sdcat/cluster/utils.py index 638fb8d..1f35ee7 100644 --- a/sdcat/cluster/utils.py +++ b/sdcat/cluster/utils.py @@ -110,7 +110,7 @@ def crop_square_image(row, square_dim: int): if Path(row.crop_path).exists(): # If the crop already exists, skip it return - print("Image widht ", row.image_width) + x1 = int(row.image_width * row.x) y1 = int(row.image_height * row.y) x2 = int(row.image_width * row.xx) diff --git a/sdcat/config/config.ini b/sdcat/config/config.ini index ef3a1a6..8528c70 100644 --- a/sdcat/config/config.ini +++ b/sdcat/config/config.ini @@ -37,8 +37,8 @@ min_similarity = 0.70 # dino_vits8 has block_size=8 which can be good for very small objects # dino_vits14 has block_size=14 # Smaller block_size means more patches and more accurate fine-grained clustering on smaller objects -model = dino_vits8 -;model = dinov2_vits14 +;model = dino_vits8 +model = dinov2_vits14 ;model = dinov2_vitb14 [detect]