3bba21d83b
refactor: merge _check_fault and _classify_fault into _analyze_fault
...
一次调用返回(fault_status, fault_type)元组，简化诊断逻辑
2026-03-27 12:10:27 +08:00
085cbc9950
fix: optimize diagnosis output format and fix inconsistency between fault_status and fault_type
...
1. 输出格式优化:
- faults中移除冗余的bbox和confidence字段
- summary中fault_count改名为detection_count
2. fault_status与fault_type逻辑统一:
- normal状态时fault_type为'none'，不再使用'suspect'
- abnormal状态时才进行故障类型分类(hotspot/local_overheating/overheat)
3. 修复usage_example.py图像通道问题:
- YOLO检测使用彩色图，温度提取时再转灰度图
2026-03-27 12:03:48 +08:00
febd9eea8d
feat: add device parameter support for GPU/CPU switching in inference API
...
- InfraredDetector.detect() 和 detect_and_draw() 新增 device 参数
- TempFeatureExtractor.detect_and_extract() 新增 device 参数
- FaultDiagnoser.diagnose() 和 batch_diagnose() 新增 device 参数
- batch_extract_features() 新增 device 参数
- 命令行 --device 选项支持指定推理设备
- detect_and_draw() 返回值改为 (image, n_detections) 避免重复检测
2026-03-27 09:10:57 +08:00
10fb804c14
fix: fix image duplicate loading in diagnose function
...
将cv2.imread从循环内移至循环前，避免对同一张图片重复加载
2026-03-27 08:51:59 +08:00
fd1a5f4050
chore: initial commit for Infrared Equipment Detection System
...
- Add YOLO-based infrared thermal image detection system
- Add dataset configuration (dataset.yaml) with 12 equipment classes
- Add training, validation and detection scripts
- Add temperature feature extraction and fault diagnosis modules
- Update README and CLAUDE.md with correct dataset paths
- Optimize .gitignore to track dataset.yaml while ignoring large data files
2026-03-26 21:36:08 +08:00
SkyLi