feat: 特殊车牌生成管线 + C++/多进程/色温校正

CCPD_SPE_GEN/cpp_pipeline/plate_swap.hpp

@@ -0,0 +1,80 @@
+#pragma once
+#include <opencv2/opencv.hpp>
+#include <string>
+#include <vector>
+#include <random>
+#include <unordered_map>
+
+namespace ice {
+
+// ── CCPD 角点解析 ──
+struct CCPDMeta {
+    cv::Point2f corners[4];  // TL, TR, BR, BL
+    bool valid = false;
+};
+CCPDMeta parse_ccpd_filename(const std::string& filename);
+
+// ── 合成车牌池 ──
+struct PlateEntry {
+    cv::Mat image;              // BGR
+    std::string plate_number;   // 车牌号 (e.g. "粤Z1234港")
+};
+class PlatePool {
+public:
+    void load(const std::string& dir);
+    void load_bin(const std::string& bin_path);
+    const PlateEntry& random(const std::string& plate_type, std::mt19937& rng);
+    size_t count(const std::string& plate_type) const;
+private:
+    std::unordered_map<std::string, std::vector<PlateEntry>> pools_;
+};
+
+// ── 管线核心 ──
+struct SwapConfig {
+    std::string plate_type;
+    std::string output_dir;
+    int h_shift = 0, s_scale = 100, v_scale = 100;
+    bool fast_mode = false;
+};
+
+// 单张处理 (pre-parsed meta 避免重复解析)
+bool process_single(const std::string& img_path,
+                    const SwapConfig& cfg,
+                    PlatePool& pool,
+                    std::mt19937& rng,
+                    const CCPDMeta* pre_meta = nullptr);
+
+// 透视提取车牌ROI
+cv::Mat extract_plate_roi(const cv::Mat& img, const cv::Point2f corners[4],
+                          int plate_w, int plate_h);
+
+// 色温匹配
+cv::Mat match_plate_color(const cv::Mat& synthetic, const cv::Mat& orig_roi,
+                          const std::string& bg_color);
+
+// 底板色校正
+cv::Mat fix_bg_color(const cv::Mat& plate, const std::string& bg_color,
+                     int h_shift, int s_scale, int v_scale);
+
+// 透视回贴: 返回 (warped, mask)
+std::pair<cv::Mat, cv::Mat> warp_plate_back(const cv::Mat& plate,
+                                             const cv::Point2f dst_corners[4],
+                                             cv::Size output_size);
+
+// 羽化贴合
+cv::Mat feather_blend(const cv::Mat& original, const cv::Mat& warped,
+                      const cv::Mat& mask, int edge_width = 8);
+
+// 噪声匹配
+cv::Mat match_noise(const cv::Mat& blended, const cv::Mat& original,
+                    const cv::Mat& mask);
+
+// 退化模拟
+cv::Mat degrade_plate(const cv::Mat& plate);
+
+// ── 工具 ──
+cv::Mat imread_unicode(const std::string& path);
+bool imwrite_unicode(const std::string& path, const cv::Mat& img,
+                     const std::vector<int>& params = {});
+
+} // namespace ice