抽烟识别模型训练的方法及装置、电子设备和存储介质
摘要:
本说明书一个或多个实施例公开了一种抽烟识别模型训练的方法及装置、电子设备和存储介质。所述方法包括:将训练集中的训练图像输入抽烟识别模型的区域划分子模型,基于区域划分子模型从训练图像中划分出因果特征区域;将训练图像输入抽烟识别模型的特征提取子模型,基于表征模块对训练图像中的特征提取,得到基础特征,以及基于因果模块对因果特征区域之间的特征提取,得到关联特征;对训练图像对应的关联特征和基础特征进行特征融合处理,得到融合特征,并将融合特征输入抽烟识别模型的分类器,得到针对训练图像的抽烟行为的识别结果;基于训练图像对应的识别结果和训练图像的标签信息,对抽烟识别模型进行训练,得到训练后的抽烟识别模型。
0/0