Commit Graph

8 Commits

Author SHA1 Message Date
SkyLi 10fb804c14 fix: fix image duplicate loading in diagnose function
将cv2.imread从循环内移至循环前,避免对同一张图片重复加载
2026-03-27 08:51:59 +08:00
SkyLi 70d624ac1d fix: correct project structure and file path errors in README
- 修正项目根目录名称: 1BaseOnYOLO → 2BaseOnYOLO_InfraredData
- 移除不存在的脚本: download_dataset.py, retry_download.py
- 添加存在的脚本: analyze_temp_features.py
- 统一模型权重文件名: infrared_yolov8n.pt → best.pt
2026-03-27 08:08:59 +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 fc4a2eb674 fix: remove inline comments from .gitignore patterns
Inline comments are not supported in gitignore - # only acts as a
comment when it appears at the beginning of a line. This caused
output/ to not be ignored because the pattern was matched literally.
2026-03-27 00:46:55 +08:00
SkyLi 45fd5c5663 chore: update dataset config for 12-class training
- 新数据集:12类电力设备红外图像(bushing, arrester, breaker等)
- 训练集:6735张,验证集:765张(删除2张越界标签图像)
- YOLOv8n模型训练50 epochs
- 验证结果:mAP@0.5=76.42%, mAP@0.5:0.95=61.64%
- 更新PROGRESS.md记录
2026-03-27 00:16:21 +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