Amould commited on
Commit
63c8324
·
1 Parent(s): e166f1a

Update codes.py

Browse files
Files changed (1) hide show
  1. codes.py +1 -1
codes.py CHANGED
@@ -40,7 +40,7 @@ def prepare_image_dino(image, smaller_edge_size=448, square=True, patch_size=14)
40
  return image_tensor, grid_size
41
 
42
 
43
- def extract_dino_features(img_path_list, smaller_edge_size=448):
44
  no_images = len(img_path_list)
45
  for i in range(no_images):
46
  #stack_image_batch = torch.zeros(no_images, , )
 
40
  return image_tensor, grid_size
41
 
42
 
43
+ def extract_dino_features(img_path_list, dino_model, smaller_edge_size=448):
44
  no_images = len(img_path_list)
45
  for i in range(no_images):
46
  #stack_image_batch = torch.zeros(no_images, , )