train: 更新类别权重设置

- 从配置中移除了特定于绝缘子的类别权重设置 - 替换为更通用的类别权重配置，增加了一个类别权重- 保持了权重设置的注释形式，以便未来可能的启用
2025-05-29 20:23:21 +08:00 · 2025-05-29 20:23:21 +08:00 · 21d190a658
commit 21d190a658
parent 66d0877cfd
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -31,7 +31,7 @@ TRAIN_CONFIG = {
        "weight_decay": 0.001,     # 权重衰减
        "label_smoothing": 0.1,    # 标签平滑
        "mosaic": True,            # 启用Mosaic增强
-        # "class_weights": [1.0, 2.0]  # 设置类别权重，insulator权重更高
+        # "class_weights": [2.0, 2.0, 1.0]  # 设置类别权重
    },
    "stage2": {
        "epochs": 150,              # 增加训练轮数