HReynaud commited on
Commit
3906a56
·
1 Parent(s): 63a37a5
Files changed (1) hide show
  1. demo.py +4 -4
demo.py CHANGED
@@ -179,8 +179,8 @@ reid = {
179
  },
180
  "tau": {
181
  "A4C": 0.9997,
182
- "PSAX": 0.9953,
183
- "PLAX": 0.9950,
184
  },
185
  }
186
 
@@ -403,12 +403,12 @@ def check_privacy(latent_image_numpy, class_selection):
403
  if corr > tau:
404
  return (
405
  None,
406
- f"⚠️ **Warning:** Generated image is too similar to training data. Privacy check failed (corr = {corr:.4f} / tau = {tau:.4f})",
407
  )
408
  else:
409
  return (
410
  latent_image_numpy,
411
- f"✅ **Success:** Generated image passed privacy check (corr = {corr:.4f} / tau = {tau:.4f})",
412
  )
413
 
414
 
 
179
  },
180
  "tau": {
181
  "A4C": 0.9997,
182
+ "PSAX": 0.9997,
183
+ "PLAX": 0.9997,
184
  },
185
  }
186
 
 
403
  if corr > tau:
404
  return (
405
  None,
406
+ f"⚠️ **Warning:** Generated image is too similar to training data. Privacy check failed.",
407
  )
408
  else:
409
  return (
410
  latent_image_numpy,
411
+ f"✅ **Success:** Generated image passed privacy check.",
412
  )
413
 
414