refactor(data): 更新数据集路径并调整数据验证逻辑
- 修改 data.yaml 中的训练、验证和测试数据集路径 - 更新 train.py 中的数据验证逻辑,使用新的数据集路径
This commit is contained in:
parent
b4468c7808
commit
66d0877cfd
@ -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']
|
||||
|
6
train.py
6
train.py
@ -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():
|
||||
|
Loading…
Reference in New Issue
Block a user