ongkn commited on
Commit
c6c02a4
·
unverified ·
1 Parent(s): 6b67e23

forgot to set up resizing since with gradcam, it must be done explicitly

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ def classify_image(input):
33
  if face is None:
34
  return "No face detected", 0, input
35
  face = Image.fromarray(face)
 
36
  tensor = transforms.ToTensor()(face)
37
  dffImage = run_dff_on_image(model=model,
38
  target_layer=targetLayerDff,
 
33
  if face is None:
34
  return "No face detected", 0, input
35
  face = Image.fromarray(face)
36
+ face = face.resize((224, 224))
37
  tensor = transforms.ToTensor()(face)
38
  dffImage = run_dff_on_image(model=model,
39
  target_layer=targetLayerDff,