ar07xd commited on
Commit
605e747
·
verified ·
1 Parent(s): a3be61c

Sync from GitHub via hub-sync

Browse files
Files changed (1) hide show
  1. services/image_service.py +6 -0
services/image_service.py CHANGED
@@ -272,6 +272,12 @@ def classify_image(
272
  total = w_ffpp + w_vit + w_eff
273
  ensemble_prob = (w_ffpp * ffpp_fake_prob + w_vit * vit_fake_prob + w_eff * eff_fake_prob) / total
274
  method = "weighted_ffpp_vit_eff"
 
 
 
 
 
 
275
  elif face_present and eff_fake_prob is not None:
276
  ensemble_prob = 0.5 * vit_fake_prob + 0.5 * eff_fake_prob
277
  method = "average_vit_eff"
 
272
  total = w_ffpp + w_vit + w_eff
273
  ensemble_prob = (w_ffpp * ffpp_fake_prob + w_vit * vit_fake_prob + w_eff * eff_fake_prob) / total
274
  method = "weighted_ffpp_vit_eff"
275
+ elif face_present and ffpp_fake_prob is not None and eff_fake_prob is None:
276
+ w_ffpp = settings.FFPP_WEIGHT_FACE
277
+ w_vit = settings.VIT_WEIGHT_FACE
278
+ total = w_ffpp + w_vit
279
+ ensemble_prob = (w_ffpp * ffpp_fake_prob + w_vit * vit_fake_prob) / total
280
+ method = "weighted_ffpp_vit"
281
  elif face_present and eff_fake_prob is not None:
282
  ensemble_prob = 0.5 * vit_fake_prob + 0.5 * eff_fake_prob
283
  method = "average_vit_eff"