基于目标检测的计数模型处理方法、装置及计算机设备
摘要:
本申请实施例属于人工智能领域,涉及一种基于目标检测的计数模型处理方法、装置、计算机设备及存储介质,方法包括:获取带有标签的原始图像;将原始图像输入初始计数模型,以通过模型的第一分支网络生成密度图,通过第二分支网络生成注意力图和人群边界图;将注意力图与人群边界图进行融合得到修正注意力图;基于标签,生成标准密度图、标准注意力图和标准人群边界图;基于密度图和标准密度图、修正注意力图和标准注意力图,以及人群边界图和标准人群边界图计算联合损失;根据联合损失调整初始计数模型,直至联合损失满足训练停止条件,得到计数模型。此外,本申请还涉及区块链技术,训练样本集可存储于区块链中。本申请提高了计数模型的准确性。
0/0