4 个月前

一种具有迭代原型适应的低样本目标计数网络

一种具有迭代原型适应的低样本目标计数网络

摘要

我们研究了在图像中对任意语义类别进行少量样本(低样本)计数的问题,仅使用少数标注示例(少样本)或不使用任何示例(零样本)。传统的少样本流程包括从示例中提取外观查询,并将其与图像特征匹配以推断物体数量。现有方法通过特征池化提取查询,这忽略了形状信息(例如,大小和长宽比),导致物体定位精度降低和计数估计不准确。为此,我们提出了一种具有迭代原型适应功能的低样本物体计数网络(LOCA)。我们的主要贡献在于新的物体原型提取模块,该模块通过迭代融合示例的形状和外观信息与图像特征来提高准确性。该模块可以轻松适应零样本场景,从而使LOCA能够涵盖整个低样本计数问题的范围。在FSC147基准测试中,LOCA在单样本和少样本任务上的均方根误差(RMSE)比最近的所有先进方法提高了20-30%,并在零样本场景中达到了最先进的水平,同时展示了更好的泛化能力。

代码仓库

djukicn/loca
官方
pytorch

基准测试

基准方法指标
exemplar-free-counting-on-fsc147LOCA
MAE(test): 16.2
MAE(val): 17.43
RMSE(test): 103.96
RMSE(val): 54.96
object-counting-on-fsc147LOCA
MAE(test): 10.79
MAE(val): 10.24
RMSE(test): 56.97
RMSE(val): 32.56

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
一种具有迭代原型适应的低样本目标计数网络 | 论文 | HyperAI超神经