• v0.2.0 8813cdf6f1

    SkyLiCoder 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