HVC-Tunnel-Multimodal/Infrared-ai-research-lab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity
24 Commits 1 Branch 2 Tags
main
Commit Graph

9 Commits

Author SHA1 Message Date
SkyLi 8813cdf6f1 refactor: decouple TempFeatureExtractor from detection logic 
- Remove detect_and_extract() method from TempFeatureExtractor
  (moved to FaultDiagnoser pipeline)
- Remove YOLO model loading from TempFeatureExtractor.__init__
- Remove duplicate train_yolo_model/validate_model from detect.py
  (already exist in scripts/train.py)
- Fix usage_example.py to use new import path and TempFeatureExtractor API
 2026-03-29 00:33:11 +08:00
SkyLi b4308d6cdc fix: update analyze_temp_features.py for new API 
- 移除对InfraredDetector的依赖，改用TempFeatureExtractor.detect_and_extract()
- 添加--device参数支持GPU/CPU切换
- 简化特征提取逻辑，直接使用detect_and_extract()返回的特征
- 添加空结果警告提示
 2026-03-27 12:21:22 +08:00
SkyLi 60b5b89a7c chore: remove legacy convert_to_yolo.py script 
新项目数据集已是完整标注的YOLO格式，无需旧的数据转换工具
 2026-03-27 12:14:54 +08:00
SkyLi 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
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 ff3dbbfd48 refactor: extract repeated sample image path to module constant 
Replace 6 inline string repetitions of the sample image path with
a single SAMPLE_IMAGE module-level constant for easier maintenance.
 2026-03-27 01:05:50 +08:00
SkyLi 787d17a870 chore: update project docs and scripts for new dataset 
- 更新README.md数据集描述（12类，6735训练图+765验证图）
- 更新模型性能指标（新12类模型结果）
- 删除已过时的download_dataset.py和retry_download.py脚本
- 更新usage_example.py使用正确的路径和模型名
 2026-03-27 00:56:10 +08:00
SkyLi 11c5e062c2 feat: add ROI temperature feature histogram analysis script 
Analyzes validation set images using YOLO detection, extracts ROI
temperature features, and plots histograms.
 2026-03-27 00:47:33 +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