44 lines
1.6 KiB
Bash
44 lines
1.6 KiB
Bash
#!/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
|