基于注意力机制的自动计数方法

本发明公开了基于注意力机制的自动计数方法，包括：基于Transfomer注意力机制生成预测框；对生成的预测框进行筛选；并对筛选后的预测框进行聚类处理；选择出数量最多的类进一步筛选，得到示例物体；基于示例物体的边界框位置信息和框内的特征信息，对物体实现自动计数。本发明提出的方法具有较强的通用性，可应用于常见类型的物体定位与计数，在仅有少量标注实例的情况下，准确计数新类别的物体，为计算机视觉中的计数任务提供了更为通用和高效的解决方案。