refactor(data): 更新数据集路径并调整数据验证逻辑

- 修改 data.yaml 中的训练、验证和测试数据集路径
- 更新 train.py 中的数据验证逻辑,使用新的数据集路径
This commit is contained in:
Asuka 2025-05-29 20:16:02 +08:00
parent b4468c7808
commit 66d0877cfd
2 changed files with 6 additions and 6 deletions

View File

@ -1,6 +1,6 @@
train: ../train/images
val: ../valid/images
test: ../test/images
train: datasets/train/images
val: datasets/valid/images
test: datasets/test/images
nc: 3
names: ['button-off', 'button-on', 'gauge']

View File

@ -48,7 +48,7 @@ TRAIN_CONFIG = {
def enhanced_data_check():
"""增强型数据验证(稳定版)"""
print("\n=== 执行增强数据验证 ===")
label_files = glob.glob("datasets/insulator/labels/*.txt")
label_files = glob.glob("datasets/train/labels/*.txt")
class_dist = {}
# 类别分布分析
@ -63,8 +63,8 @@ def enhanced_data_check():
# 样本可视化(示例保留结构)
sample_count = 3
for i in range(sample_count):
img_path = f"datasets/insulator/images/{i:04d}.jpg"
lbl_path = f"datasets/insulator/labels/{i:04d}.txt"
img_path = f"datasets/train/images/{i:04d}.jpg"
lbl_path = f"datasets/train/labels/{i:04d}.txt"
# 添加实际可视化逻辑
def main():