feat: 特殊车牌生成管线 + C++/多进程/色温校正
This commit is contained in:
1 parent
3808b9f035
commit
3101992f51
29 files changed
+2336
-80
No files matched your search
@@ -0,0 +1,43 @@
|
||||
#!/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
|
||||
Reference in new issue
Block a user