viep-fault-detection / thermal_service.py
Sanjayraju30's picture
Create thermal_service.py
3b01344 verified
raw
history blame
359 Bytes
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