低显存高精度
一种新型空间引导高分辨率查询模块,通过从低分辨率特征中查询最近隐式嵌入,实现在严格GPU内存限制下的高效语义分割。
Qi, Li · 蔡加欣 · Jiexin, Luo · 于元隆 · Jason, Gu · Jia, Pan · 刘文犀
IEEE Robotics and Automation Letters 2024
通过空间引导的高分辨率查询模块,从低分辨率特征中查询最近隐式嵌入,显著降低 GPU 内存占用。
在消耗相近内存的条件下,mIoU 相比现有最优模型提升约 3%。
模型可在 Jetson TX2 等内存不足 8G 的嵌入式平台上部署,无需高端 GPU。