解决WD14标签下划线丢失难题：BooruDatasetTagManager全配置指南

解决WD14标签下划线丢失难题：BooruDatasetTagManager全配置指南

【免费下载链接】BooruDatasetTagManager 项目地址: https://gitcode.com/gh_mirrors/bo/BooruDatasetTagManager

在AI绘画模型训练中，标签格式的准确性直接影响训练效果。许多用户反馈使用BooruDatasetTagManager时，WD14（Waifu Diffusion 1.4）标签中的下划线会被自动替换为空格，导致masterpiece_quality变成masterpiece quality等问题，严重影响标签语义完整性。本文将从根本原因出发，提供三种解决方案，帮助用户彻底解决下划线丢失问题，同时兼容不同使用场景的需求。

问题根源：标签标准化机制的双刃剑

BooruDatasetTagManager为提升标签规范性，默认启用了标签修复功能。该机制通过多个代码层面共同作用：

核心代码位于TagsDB.cs的标签预处理逻辑：

private string PrepareTag(string tag)
{
    if (string.IsNullOrWhiteSpace(tag))
        return tag;
    if (FixTags)  // 受FixTagsOnSaveLoad控制
    {
        tag = tag.Replace('_', ' ');  // 下划线替换为空格
        tag = tag.Replace("\\(", "(");
        tag = tag.Replace("\\)", ")");
    }
    return tag;
}

同时在PromptParser.cs的权重解析过程中存在二次处理：

// 标签分割时的下划线替换
string textTag = tag.Replace('_', ' ').ToLower().Trim();

这种设计虽然能统一标签格式，但破坏了WD14标签体系中以下划线区分复合概念的命名规范（如1girl_with_cat_ears）。

解决方案一：图形界面快速配置（推荐普通用户）

BooruDatasetTagManager在v1.0.3版本后提供了可视化开关，通过3步即可禁用下划线替换：

1. 打开程序主界面，点击顶部菜单栏Settings（设置）
2. 在弹出的设置窗口中切换到General（通用）选项卡
3. 取消勾选Fix tags on load/save选项，其完整描述为：

   "Automatically replace underscores with spaces and fix parentheses escaping when loading and saving tags"

配置验证：修改后保存设置，导入包含下划线的标签文件，通过以下方法确认：

• 查看标签编辑框中的文本是否保留下划线
• 检查导出的标签文件（.txt/.csv）原始内容
• 通过"Preview"功能观察标签渲染效果

解决方案二：高级用户的配置文件修改

对于需要批量部署或无法通过UI操作的场景，可直接修改配置文件：

1. 定位配置文件settings.json，通常位于：

   • Windows: C:\Users\[用户名]\AppData\Roaming\BooruDatasetTagManager\
   • Linux: ~/.config/BooruDatasetTagManager/
   • macOS: ~/Library/Application Support/BooruDatasetTagManager/

2. 添加或修改以下配置项：

{
  "FixTagsOnSaveLoad": false,
  "SeparatorOnLoad": ",",
  "SeparatorOnSave": ", "
}

3. 保存文件并重启程序，配置将立即生效

批量部署技巧：企业用户可通过组策略或脚本将此配置推送到多台设备，命令示例：

# Linux/macOS批量配置命令
sed -i 's/"FixTagsOnSaveLoad": true/"FixTagsOnSaveLoad": false/' ~/.config/BooruDatasetTagManager/settings.json

解决方案三：开发者的代码级定制

对于需要部分保留下划线（如仅保留特定前缀标签的下划线）的高级场景，需修改核心处理逻辑：

修改标签预处理逻辑

在TagsDB.cs中添加条件判断，仅对特定标签保留下划线：

private string PrepareTag(string tag)
{
    if (string.IsNullOrWhiteSpace(tag))
        return tag;
    if (FixTags)
    {
        // 保留以"wd14_"开头的标签下划线
        if (!tag.StartsWith("wd14_"))
        {
            tag = tag.Replace('_', ' ');
        }
        tag = tag.Replace("\\(", "(");
        tag = tag.Replace("\\)", ")");
    }
    return tag;
}

调整权重解析器行为

修改PromptParser.cs中的标签分割逻辑：

// 原代码：强制替换所有下划线
string textTag = tag.Replace('_', ' ').ToLower().Trim();

// 修改为：可选保留下划线
string textTag = FixTags ? tag.Replace('_', ' ').ToLower().Trim() : tag.ToLower().Trim();

添加自定义配置项

在AppSettings.cs中添加精细化控制选项：

public class AppSettings
{
    // 现有配置...
    public bool FixTagsOnSaveLoad { get; set; } = true;
    public bool PreserveUnderscoresInWD14 { get; set; } = false;  // 新增配置
    public string UnderscorePreservePattern { get; set; } = "^wd14_.*";  // 正则匹配模式
}

编译部署：修改后通过以下命令重新构建项目：

dotnet build BooruDatasetTagManager.sln -c Release

三种方案的对比与选择建议

方案类型	操作难度	适用场景	优点	缺点
图形界面配置	⭐⭐⭐⭐⭐	普通用户、临时需求	操作简单、即时生效、无风险	全局生效，无法精细化控制
配置文件修改	⭐⭐⭐	批量部署、服务器环境	可脚本化、无需重新编译	无UI反馈，易输错格式
代码级定制	⭐	开发人员、特殊需求	高度自定义、支持复杂规则	需维护代码分支、有兼容性风险

决策流程图：

常见问题与解决方案

Q1: 禁用标签修复后出现重复标签怎么办？

A: 这是因为原标签系统中存在masterpiece和masterpiece_quality等相似标签。建议：

1. 使用"Replace All"功能合并相似标签
2. 在TagsDB.cs中添加去重逻辑：

// 添加标签前检查是否已存在
if (Tags.Exists(a => a.Parent == tag))
    return;

Q2: 配置不生效如何排查？

A: 按以下步骤诊断：

1. 确认settings.json中FixTagsOnSaveLoad值为false
2. 检查程序版本是否≥v1.0.3（旧版本无此配置项）
3. 验证标签文件编码是否为UTF-8（非UTF-8可能导致解析异常）
4. 查看日志文件（%APPDATA%\BooruDatasetTagManager\logs）中的错误信息

Q3: 能否同时保留下划线和自动修复括号？

A: 可修改PrepareTag方法实现部分修复：

if (FixTags)
{
    // 仅修复括号，保留下划线
    tag = tag.Replace("\\(", "(");
    tag = tag.Replace("\\)", ")");
    // 移除下划线替换逻辑
    // tag = tag.Replace('_', ' ');
}

总结与最佳实践

保留WD14标签下划线的核心是控制FixTagsOnSaveLoad配置项，普通用户推荐使用图形界面配置，开发者可通过代码定制实现更灵活的标签处理策略。最佳实践建议：

1. 版本控制：使用Git管理自定义代码，便于同步官方更新
2. 测试验证：修改配置后通过"Preview"功能批量检查标签格式
3. 文档记录：对自定义规则进行详细注释，方便后续维护
4. 社区反馈：通过项目GitHub Issues分享使用场景，帮助官方改进功能

通过本文介绍的方法，用户可彻底解决WD14标签下划线丢失问题，同时根据实际需求选择最适合的实现方案。正确的标签格式处理将为AI绘画模型训练提供更精准的训练数据，显著提升最终生成效果的质量与一致性。

本文基于BooruDatasetTagManager v1.2.0版本编写，不同版本间可能存在差异。建议定期关注项目更新日志，及时获取功能改进信息。

【免费下载链接】BooruDatasetTagManager 项目地址: https://gitcode.com/gh_mirrors/bo/BooruDatasetTagManager