Spaces:
Runtime error
Runtime error
canadianjosieharrison
commited on
Commit
•
8b87a69
1
Parent(s):
b45ac3d
Update similarity_inference.py
Browse files- similarity_inference.py +2 -2
similarity_inference.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
from image_helpers import convert_images_to_grayscale,
|
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 |
-
|
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/"
|