#!/bin/bash
# ============================================================
# GAN 数据准备：合成车牌(A) + 真实车牌(B)
# 用法: cd ICE_Dataset && bash CCPD_SPE_GEN/scripts/gan_data_prep.sh
# ============================================================
source /home/huawei/rknn_env/bin/activate  # TODO: 替换为你的虚拟环境路径
cd /mnt/d/ICEv2/ICE_Dataset               # TODO: 替换为你的 ICE_Dataset 路径

OUT_A=./gan_data/domain_A_synthetic
OUT_B=./gan_data/domain_B_real
CCPD=./CCPD/CCPD2019/ccpd_base,./CCPD/CCPD2019/ccpd_tilt,./CCPD/CCPD2019/ccpd_rotate,./CCPD/CCPD2019/ccpd_fn

echo "============================================"
echo " Phase 1: 生成合成车牌 (9类型 x 1000张)"
echo "============================================"
for p in police army gangao shiguan_old shiguan_new lingguan_old lingguan_new coach new_energy; do
    echo "--- $p ---"
    python3 CCPD_SPE_GEN/cv_pipeline/plate_swap.py \
        --ccpd-dir ./CCPD/CCPD2019/ccpd_base \
        --plate-type $p \
        --num 1000 \
        --output $OUT_A/$p \
        --seed 42 2>&1 | tail -2
done

echo ""
echo "============================================"
echo " Phase 2: 提取真实车牌ROI (1000张)"
echo "============================================"
python3 CCPD_SPE_GEN/cv_pipeline/extract_real_plates.py \
    --ccpd-dir "$CCPD" \
    --num 1000 \
    --output $OUT_B \
    --seed 42

echo ""
echo "============================================"
echo " GAN 数据准备完成!"
echo "  Domain A (合成): $OUT_A"
echo "  Domain B (真实): $OUT_B"
echo "============================================"
ls -l $OUT_A/*/ | wc -l
ls -l $OUT_B/ | wc -l