#!/bin/bash

# 图像质量优化系统运行脚本

echo "=== 图像质量优化系统 ==="
echo "请选择运行模式："
echo "1. 测试模式 (快速验证功能)"
echo "2. 完整优化 (处理所有图像)"
echo "3. 限制优化 (处理前100张低质量图像)"

read -p "请输入选择 (1-3): " choice

case $choice in
    1)
        echo "启动测试模式..."
        cd /home/ubuntu/lyl/QwenIllustrious/data_tool/improve_data_quality
        python test_optimizer.py
        ;;
    2)
        echo "启动完整优化模式..."
        cd /home/ubuntu/lyl/QwenIllustrious/data_tool/improve_data_quality
        python quality_optimizer.py
        ;;
    3)
        echo "启动限制优化模式..."
        cd /home/ubuntu/lyl/QwenIllustrious/data_tool/improve_data_quality
        python -c "
from quality_optimizer import QualityOptimizer
optimizer = QualityOptimizer()
optimizer.run_full_optimization(batch_size=20, max_regenerate=100)
"
        ;;
    *)
        echo "无效选择"
        exit 1
        ;;
esac

echo "处理完成！"