Commit Graph

4 Commits

Author SHA1 Message Date
SkyLi 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
SkyLi 9a5f6fe473 fix: fix multiple image duplicate loading issues
- detect.py: detect_and_draw()中传给detect()的是已加载的img数组而非字符串路径
- usage_example.py: example3中复用gray_img数组而非重新加载
- analyze_temp_features.py: 复用已加载的gray_img数组进行检测
2026-03-27 08:53:29 +08:00
SkyLi 10fb804c14 fix: fix image duplicate loading in diagnose function
将cv2.imread从循环内移至循环前,避免对同一张图片重复加载
2026-03-27 08:51:59 +08:00
SkyLi 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