File size: 3,995 Bytes
14cb7ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
# models/image_analysis.py
from PIL import Image
import numpy as np
from transformers import AutoImageProcessor, AutoModelForImageClassification
from .logging_config import logger
# Initialize real estate classification model
try:
processor = AutoImageProcessor.from_pretrained("andupets/real-estate-image-classification")
model = AutoModelForImageClassification.from_pretrained("andupets/real-estate-image-classification")
has_model = True
logger.info("Real estate classification model loaded successfully")
except Exception as e:
logger.error(f"Error loading real estate classification model: {str(e)}")
has_model = False
def analyze_image(image):
try:
if has_model:
img_rgb = image.convert('RGB')
inputs = processor(images=img_rgb, return_tensors="pt")
outputs = model(**inputs)
logits = outputs.logits
probs = logits.softmax(dim=1).detach().numpy()[0]
# Get the highest confidence prediction
max_prob_idx = probs.argmax()
max_prob = probs[max_prob_idx]
predicted_label = model.config.id2label[max_prob_idx]
# Check if it's a real estate image (confidence > 0.5)
is_real_estate = max_prob > 0.5
quality = assess_image_quality(image)
is_ai_generated = detect_ai_generated_image(image)
return {
'is_property_related': is_real_estate,
'property_confidence': float(max_prob),
'predicted_label': predicted_label,
'top_predictions': [
{'label': model.config.id2label[i], 'confidence': float(prob)}
for i, prob in enumerate(probs)
],
'image_quality': quality,
'is_ai_generated': is_ai_generated,
'authenticity_score': 0.95 if not is_ai_generated else 0.60
}
else:
logger.warning("Real estate classification model unavailable")
return {
'is_property_related': False,
'property_confidence': 0.0,
'predicted_label': 'unknown',
'top_predictions': [],
'image_quality': assess_image_quality(image),
'is_ai_generated': False,
'authenticity_score': 0.5
}
except Exception as e:
logger.error(f"Error analyzing image: {str(e)}")
return {
'is_property_related': False,
'property_confidence': 0.0,
'predicted_label': 'error',
'top_predictions': [],
'image_quality': {'resolution': 'unknown', 'quality_score': 0},
'is_ai_generated': False,
'authenticity_score': 0.0,
'error': str(e)
}
def detect_ai_generated_image(image):
try:
img_array = np.array(image)
if len(img_array.shape) == 3:
gray = np.mean(img_array, axis=2)
else:
gray = img_array
noise = gray - np.mean(gray)
noise_std = np.std(noise)
width, height = image.size
perfect_dimensions = (width % 64 == 0 and height % 64 == 0)
has_exif = hasattr(image, '_getexif') and image._getexif() is not None
return noise_std < 0.05 or perfect_dimensions or not has_exif
except Exception as e:
logger.error(f"Error detecting AI-generated image: {str(e)}")
return False
def assess_image_quality(img):
try:
width, height = img.size
resolution = width * height
quality_score = min(100, resolution // 20000)
return {
'resolution': f"{width}x{height}",
'quality_score': quality_score
}
except Exception as e:
logger.error(f"Error assessing image quality: {str(e)}")
return {
'resolution': 'unknown',
'quality_score': 0
}
|