-
模块解耦与结构优化 Stable
released this
2026-03-29 00:41:35 +08:00 | 2 commits to main since this release主要更新
代码解耦
TempFeatureExtractor不再依赖 YOLO 模型,专注温度特征提取InfraredDetector专注目标检测FaultDiagnoser负责流程编排- 移除
detect.py中重复的train_yolo_model/validate_model函数
包结构优化
- 重构为标准
src-layout包结构 - 导出
FaultDiagnoser,InfraredDetector,TempFeatureExtractor三个核心类 - 简化 API 调用方式
使用示例
from infrared_data import InfraredDetector, TempFeatureExtractor, FaultDiagnoser # 温度特征提取(无需模型) extractor = TempFeatureExtractor() features = extractor.extract_from_roi(roi) # 故障诊断 diagnoser = FaultDiagnoser("weights/best.pt") result = diagnoser.diagnose("image.jpg")升级注意
- TempFeatureExtractor 构造函数不再接受 model_path 参数
- 如需检测+温度提取流程,请使用 FaultDiagnoser.diagnose()
Downloads