公开(公告)号：CN117351301A

公开(公告)日：2024-01-05

申请号：CN202311281863.6

申请日：2023-09-27

Applicant: 鹏城实验室

Inventor： 王耀威 ,  郭明月 ,  颜肇义 ,  叶齐祥

IPC: G06V10/774 ,  G06V10/776 ,  G06V10/764 ,  G06V10/82 ,  G06V10/44 ,  G06N3/0464 ,  G06N3/084 ,  G06N3/048

Abstract: 本申请提供了一种域自适应模型训练方法与相关方法、装置、设备及介质，通过获取图像样本，进行特征提取，得到第一图像特征；对第一图像特征进行卷积处理，得到第二图像特征；对第二图像特征进行预测处理，得到预测虚拟类标，根据预测虚拟类标与目标虚拟类标的差异对参数器进行训练；对第二图像特征进行信息处理，得到分类信息；对第一图像特征进行分类处理，得到域自适应图像特征；对域自适应特征进行预测处理，得到预测密度图，根据预测密度图与真实密度图的差异对骨干网络进行训练；返回特征提取步骤进行迭代，直至差异收敛，得到训练后的域自适应模型。通过单阶段训练策略，使模型能够对不同域的数据进行分类，提升人群计数的准确性。

公开(公告)号：CN118115932A

公开(公告)日：2024-05-31

申请号：CN202311208964.0

申请日：2023-09-18

Applicant: 鹏城实验室

Inventor： 王耀威 ,  郭明月 ,  颜肇义 ,  叶齐祥

IPC: G06V20/52 ,  G06V10/26 ,  G06V10/44 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/08

Abstract: 本申请提供了一种图像回归器训练方法与相关方法、装置、设备及介质。该图像回归器训练方法包括：获取图像样本并提取图像特征；通过分割器输出第一预测分割掩码，通过图像回归器输出第一预测密度图；将第一预测密度图与第一分割掩码结合输出目标分割掩码对分割器进行提示训练；基于提示训练后的分割器输出第二预测分割掩码；通过图像回归器输出第二预测密度图；更新目标分割掩码，并返回根据目标分割掩码对分割器进行提示训练的步骤继续进行迭代互相提示学习，直至满足迭代条件，得到训练后的图像回归器。以此，通过互相提示学习的方式进行图像回归器训练，以获取到人头位置的精确空间信息，减少点标签不确定性带来的影响，提升人群计数的准确性。

公开(公告)号：CN117057392A

公开(公告)日：2023-11-14

申请号：CN202310928096.7

申请日：2023-07-26

Applicant: 鹏城实验室

Inventor： 颜肇义 ,  叶齐祥 ,  王耀威 ,  高文

IPC: G06N3/04 ,  G06N3/08

Abstract: 本申请公开了iTPN模型在MindSpore框架的适配方法、装置及存储介质，适配方法通过将训练数据转换成适于输入MindSpore框架的MindRecord文件并对转换后的数据进行数据增强，然后将iTPN模型中Pytorch的模型参数和模型函数映射到MindSpore框架中，并按照各层权重进行精度转换以适于MindSpore框架的计算精度，还设定逐层学习率，在实际训练过程中基于当前层的学习率进行优化，并实现数据并行、优化器并行和梯度合并处理，节省了GPU的计算负担，加快了适配到MindSpore框架后的计算效率，适配后的模型能够作为基础网络用于多种下游任务，从而提升下游任务的性能精度。