Spaces:
Sleeping
Sleeping
Sync from GitHub via hub-sync
Browse files
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"
|