finish Inference and Matching Part.

This commit is contained in:
Jiao77
2025-09-25 22:05:39 +08:00
parent 2ccfe7b07f
commit 419a7db543
6 changed files with 346 additions and 18 deletions

View File

@@ -62,13 +62,13 @@
> *目标:大幅提升大尺寸版图的匹配速度和多尺度检测能力。*
- [ ] **将模型改造为特征金字塔网络 (FPN) 架构**
- [x] **将模型改造为特征金字塔网络 (FPN) 架构**
- **✔️ 价值**: 当前的多尺度匹配需要多次缩放图像并推理速度慢。FPN 只需一次推理即可获得所有尺度的特征,极大加速匹配过程。
- **📝 执行方案**:
1. 修改 `models/rord.py`,从骨干网络的不同层级(如 VGG 的 `relu2_2`, `relu3_3`, `relu4_3`)提取特征图。
2. 添加上采样和横向连接层来融合这些特征图,构建出特征金字塔。
3. 修改 `match.py`,使其能够直接从 FPN 的不同层级获取特征,替代原有的图像金字塔循环。
- [ ] **在滑动窗口匹配后增加关键点去重**
- [x] **在滑动窗口匹配后增加关键点去重**
- **✔️ 价值**: `match.py` 中的滑动窗口在重叠区域会产生大量重复的关键点,增加后续匹配的计算量并可能影响精度。
- **📝 执行方案**:
1.`match.py``extract_features_sliding_window` 函数返回前。