Wolowolo commited on
Commit
8e5a9fc
·
verified ·
1 Parent(s): 3aafd1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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: