File size: 359 Bytes
3b01344
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from ultralytics import YOLO

# Load thermal model
thermal_model = YOLO("thermal_model.pt")  # You must upload your model here

def detect_thermal_anomalies(image_path):
    results = thermal_model(image_path)
    flagged = []
    for r in results:
        if hasattr(r, 'temperature') and r.temperature > 75:
            flagged.append(r)
    return flagged