Update app.py
Browse files
app.py
CHANGED
@@ -242,7 +242,7 @@ def FSFM3C_image_detection(image):
|
|
242 |
target_layers=[model.blocks[-1].norm1],
|
243 |
reshape_transform=reshape_transform
|
244 |
)
|
245 |
-
grayscale_cam = cam(input_tensor=input_tensor, targets=[target_category])
|
246 |
grayscale_cam = 1 - grayscale_cam[0, :]
|
247 |
img = np.array(img)
|
248 |
if img.shape[2] == 4:
|
|
|
242 |
target_layers=[model.blocks[-1].norm1],
|
243 |
reshape_transform=reshape_transform
|
244 |
)
|
245 |
+
grayscale_cam = cam(input_tensor=input_tensor, targets=[target_category], aug_smooth=False, eigen_smooth=True)
|
246 |
grayscale_cam = 1 - grayscale_cam[0, :]
|
247 |
img = np.array(img)
|
248 |
if img.shape[2] == 4:
|