|
|
|
|
| """
|
| 回声处理配置验证脚本
|
|
|
| 检查当前配置是否正确启用了激进的去回声处理
|
| """
|
|
|
| import json
|
| import os
|
| from pathlib import Path
|
|
|
| def check_deecho_models():
|
| """检查 DeEcho 模型是否存在"""
|
| print("=" * 60)
|
| print("检查 DeEcho 模型")
|
| print("=" * 60)
|
|
|
| models_dir = Path("assets/uvr5_weights")
|
| required_models = [
|
| "VR-DeEchoDeReverb.pth",
|
| "VR-DeEchoAggressive.pth",
|
| "VR-DeEchoNormal.pth",
|
| "onnx_dereverb_By_FoxJoy"
|
| ]
|
|
|
| all_found = True
|
| for model in required_models:
|
| model_path = models_dir / model
|
| exists = model_path.exists()
|
| status = "[OK]" if exists else "[NO]"
|
| size = ""
|
| if exists:
|
| if model_path.is_file():
|
| size_mb = model_path.stat().st_size / (1024 * 1024)
|
| size = f" ({size_mb:.1f} MB)"
|
| else:
|
| size = " (目录)"
|
|
|
| print(f"{status} {model}{size}")
|
| if not exists:
|
| all_found = False
|
|
|
| print()
|
| if all_found:
|
| print("[OK] All DeEcho models ready")
|
| else:
|
| print("[NO] Missing models, run: python tools/download_models.py")
|
|
|
| return all_found
|
|
|
| def check_config():
|
| """检查配置文件"""
|
| print("\n" + "=" * 60)
|
| print("检查配置文件")
|
| print("=" * 60)
|
|
|
| config_path = Path("configs/config.json")
|
| with open(config_path, 'r', encoding='utf-8') as f:
|
| config = json.load(f)
|
|
|
| cover_config = config.get("cover", {})
|
|
|
|
|
| checks = [
|
| ("VC 预处理模式", "vc_preprocess_mode", "auto", cover_config.get("vc_preprocess_mode")),
|
| ("源约束模式", "source_constraint_mode", "auto", cover_config.get("source_constraint_mode")),
|
| ("Karaoke 分离", "karaoke_separation", True, cover_config.get("karaoke_separation")),
|
| ("索引率", "index_rate", 0.50, cover_config.get("index_rate")),
|
| ("保护系数", "protect", 0.33, cover_config.get("protect")),
|
| ]
|
|
|
| all_correct = True
|
| for name, key, expected, actual in checks:
|
| match = actual == expected
|
| status = "[OK]" if match else "[NO]"
|
| print(f"{status} {name} ({key}): {actual} {'==' if match else '!='} {expected}")
|
| if not match:
|
| all_correct = False
|
|
|
| print()
|
| if all_correct:
|
| print("[OK] Config optimized for aggressive deecho")
|
| else:
|
| print("[WARN] Some configs not optimized")
|
|
|
| return all_correct
|
|
|
| def print_recommendations():
|
| """打印使用建议"""
|
| print("\n" + "=" * 60)
|
| print("使用建议")
|
| print("=" * 60)
|
|
|
| print("""
|
| 1. 当前配置使用自动模式：
|
| - 优先使用 UVR DeEcho 模型，缺模型时回退到算法去混响
|
| - DeEcho 质量好时跳过 blend 直接使用，避免混回原始回音
|
| - 源约束仅在去过回音的预处理下自动启用
|
|
|
| 2. 如果回声仍然明显，可以尝试：
|
| - 在 UI 中调整"索引率"（降低到 0.2-0.3）
|
| - 在 UI 中调整"保护系数"（降低到 0.2-0.25）
|
| - 使用更高质量的输入音频
|
|
|
| 3. 处理流程：
|
| 原始音频 → Karaoke 分离 → UVR DeEcho(或算法去混响) → RVC 转换 → 输出
|
|
|
| 4. 测试建议：
|
| - 选择一首有明显回声的歌曲
|
| - 查看日志中 DeEcho quality 指标
|
| - 对比处理前后的回声强度
|
| """)
|
|
|
| def main():
|
| print("\n回声处理配置验证\n")
|
|
|
| models_ok = check_deecho_models()
|
| config_ok = check_config()
|
| print_recommendations()
|
|
|
| print("\n" + "=" * 60)
|
| if models_ok and config_ok:
|
| print("[OK] System ready for testing")
|
| else:
|
| print("[WARN] Please fix issues above")
|
| print("=" * 60 + "\n")
|
|
|
| if __name__ == "__main__":
|
| main()
|
|
|
