Update app.py

app.py

@@ -108,10 +108,10 @@ def detect_ai_image(image):
                     if "human" in label_lower or "real" in label_lower:
                         human_label_idx = idx
                 
-                # 根据标签确定AI概率
+                # 修改后的标签解释逻辑
                 if human_label_idx is not None:
-                    # 如果有human标签，AI概率是1减去human概率
-                    ai_probability = 1 - float(probabilities[0][human_label_idx].item())
+                    # 反转解释，将human标签视为AI生成的指标
+                    ai_probability = float(probabilities[0][human_label_idx].item())
                 elif ai_label_idx is not None:
                     # 如果有AI标签
                     ai_probability = float(probabilities[0][ai_label_idx].item())
@@ -145,17 +145,36 @@ def detect_ai_image(image):
     # 分析图像特征
     image_features = analyze_image_features(image)
     
-    # 确定置信度级别
-    if final_ai_probability > 0.7:
+    # 应用特征权重调整AI概率
+    adjusted_probability = final_ai_probability
+    
+    # 低边缘密度通常表示AI生成
+    if image_features["edge_density"] < 0.01:
+        adjusted_probability += 0.2
+    
+    # 高纹理均匀性通常表示AI生成
+    if image_features["texture_homogeneity"] > 0.5:
+        adjusted_probability += 0.1
+    
+    # 低噪声水平通常表示AI生成
+    if image_features["noise_level"] < 0.5:
+        adjusted_probability += 0.1
+    
+    # 确保概率在0-1范围内
+    adjusted_probability = min(1.0, max(0.0, adjusted_probability))
+    
+    # 调整后的阈值判断
+    if adjusted_probability > 0.5:  # 降低AI判定阈值
         confidence_level = "高概率AI生成"
-    elif final_ai_probability < 0.3:
+    elif adjusted_probability < 0.2:  # 提高人类判定要求
         confidence_level = "高概率人类创作"
     else:
         confidence_level = "无法确定"
     
     # 构建最终结果
     final_result = {
-        "ai_probability": final_ai_probability,
+        "ai_probability": adjusted_probability,
+        "original_ai_probability": final_ai_probability,
         "confidence_level": confidence_level,
         "individual_model_results": results,
         "features": image_features