Spaces:
Runtime error
Runtime error
Update
Browse files- extract_lime.py +15 -1
extract_lime.py
CHANGED
|
@@ -34,6 +34,7 @@ def generate_lime(image_path=None, save_path=None):
|
|
| 34 |
image_path = r"data\test\Task 1\{}\{}".format(disease, image_path)
|
| 35 |
image_name = image_path.split(".")[0].split("\\")[-1]
|
| 36 |
image = Image.open(image_path).convert("RGB")
|
|
|
|
| 37 |
image = preprocess(image)
|
| 38 |
image = image.unsqueeze(0) # Add batch dimension
|
| 39 |
image = image.to(DEVICE)
|
|
@@ -68,11 +69,17 @@ def generate_lime(image_path=None, save_path=None):
|
|
| 68 |
os.makedirs(f"docs/evaluation/lime/{disease}", exist_ok=True)
|
| 69 |
# image.save(f'docs/evaluation/lime/{disease}/{image_name}.jpg')
|
| 70 |
plt.imsave(f"docs/evaluation/lime/{disease}/{image_name}.jpg", image)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
|
| 72 |
else:
|
| 73 |
image = None
|
| 74 |
print("Processing", image_path)
|
| 75 |
image = Image.open(image_path).convert("RGB")
|
|
|
|
| 76 |
image = preprocess(image)
|
| 77 |
image = image.unsqueeze(0) # Add batch dimension
|
| 78 |
image = image.to(DEVICE)
|
|
@@ -104,6 +111,12 @@ def generate_lime(image_path=None, save_path=None):
|
|
| 104 |
# os.makedirs(f"docs/evaluation/lime/{disease}", exist_ok=True)
|
| 105 |
# image.save(f'docs/evaluation/lime/{disease}/{image_name}.jpg')
|
| 106 |
plt.imsave(save_path, image)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 107 |
|
| 108 |
|
| 109 |
# start = time.time()
|
|
@@ -112,4 +125,5 @@ def generate_lime(image_path=None, save_path=None):
|
|
| 112 |
|
| 113 |
# end = time.time()
|
| 114 |
|
| 115 |
-
# print("Time taken:", end - start)
|
|
|
|
|
|
| 34 |
image_path = r"data\test\Task 1\{}\{}".format(disease, image_path)
|
| 35 |
image_name = image_path.split(".")[0].split("\\")[-1]
|
| 36 |
image = Image.open(image_path).convert("RGB")
|
| 37 |
+
width, height = image.size
|
| 38 |
image = preprocess(image)
|
| 39 |
image = image.unsqueeze(0) # Add batch dimension
|
| 40 |
image = image.to(DEVICE)
|
|
|
|
| 69 |
os.makedirs(f"docs/evaluation/lime/{disease}", exist_ok=True)
|
| 70 |
# image.save(f'docs/evaluation/lime/{disease}/{image_name}.jpg')
|
| 71 |
plt.imsave(f"docs/evaluation/lime/{disease}/{image_name}.jpg", image)
|
| 72 |
+
|
| 73 |
+
# Resize the image to the original size
|
| 74 |
+
image = Image.open(f"docs/evaluation/lime/{disease}/{image_name}.jpg")
|
| 75 |
+
image = image.resize((width, height))
|
| 76 |
+
image.save(f"docs/evaluation/lime/{disease}/{image_name}.jpg")
|
| 77 |
|
| 78 |
else:
|
| 79 |
image = None
|
| 80 |
print("Processing", image_path)
|
| 81 |
image = Image.open(image_path).convert("RGB")
|
| 82 |
+
width, height = image.size
|
| 83 |
image = preprocess(image)
|
| 84 |
image = image.unsqueeze(0) # Add batch dimension
|
| 85 |
image = image.to(DEVICE)
|
|
|
|
| 111 |
# os.makedirs(f"docs/evaluation/lime/{disease}", exist_ok=True)
|
| 112 |
# image.save(f'docs/evaluation/lime/{disease}/{image_name}.jpg')
|
| 113 |
plt.imsave(save_path, image)
|
| 114 |
+
|
| 115 |
+
# Resize the image to the original size
|
| 116 |
+
image = Image.open(save_path)
|
| 117 |
+
image = image.resize((width, height))
|
| 118 |
+
image.save(save_path)
|
| 119 |
+
|
| 120 |
|
| 121 |
|
| 122 |
# start = time.time()
|
|
|
|
| 125 |
|
| 126 |
# end = time.time()
|
| 127 |
|
| 128 |
+
# print("Time taken:", end - start)
|
| 129 |
+
|