Spaces:
Build error
Build error
Commit ·
029ab07
1
Parent(s): 29aa270
set cuda = False for grad cam
Browse files
app.py
CHANGED
|
@@ -167,7 +167,7 @@ def process_gradcam_images(num_images,layer,opacity,image_list=None):
|
|
| 167 |
selected_data = [image_list]
|
| 168 |
|
| 169 |
layer_model = dict_layer.get(layer)
|
| 170 |
-
cam = GradCAM(model=loaded_model, target_layers = [layer_model], use_cuda =
|
| 171 |
grad_images = []
|
| 172 |
inv_normalize = transforms.Normalize(
|
| 173 |
mean=[-0.50/0.2197, -0.50/0.1858, -0.50/0.1569], # mean_ds = [0.2197, 0.1858, 0.1569]
|
|
|
|
| 167 |
selected_data = [image_list]
|
| 168 |
|
| 169 |
layer_model = dict_layer.get(layer)
|
| 170 |
+
cam = GradCAM(model=loaded_model, target_layers = [layer_model], use_cuda = False)
|
| 171 |
grad_images = []
|
| 172 |
inv_normalize = transforms.Normalize(
|
| 173 |
mean=[-0.50/0.2197, -0.50/0.1858, -0.50/0.1569], # mean_ds = [0.2197, 0.1858, 0.1569]
|