canadianjosieharrison commited on
Commit
8b87a69
1 Parent(s): b45ac3d

Update similarity_inference.py

Browse files
Files changed (1) hide show
  1. similarity_inference.py +2 -2
similarity_inference.py CHANGED
@@ -1,4 +1,4 @@
1
- from image_helpers import convert_images_to_grayscale, crop_center_largest_contour, fetch_similar
2
  import datasets as ds
3
  import re
4
  import torchvision.transforms as T
@@ -33,7 +33,7 @@ def similarity_inference(directory):
33
  print(color_dict)
34
 
35
  convert_images_to_grayscale(directory)
36
- crop_center_largest_contour(directory)
37
 
38
  # define processing variables needed for embedding calculation
39
  root_directory = "data/" #"C:/Users/josie/OneDrive - Chalmers/Documents/Speckle hackathon/data/"
 
1
+ from image_helpers import convert_images_to_grayscale, crop_least_variant_patch, fetch_similar
2
  import datasets as ds
3
  import re
4
  import torchvision.transforms as T
 
33
  print(color_dict)
34
 
35
  convert_images_to_grayscale(directory)
36
+ crop_least_variant_patch(directory)
37
 
38
  # define processing variables needed for embedding calculation
39
  root_directory = "data/" #"C:/Users/josie/OneDrive - Chalmers/Documents/Speckle hackathon/data/"