智能建造目标计数方法、电子设备及存储介质
摘要:
本申请提供一种智能建造目标计数方法、电子设备及存储介质。方法包括:获取待识别图像;将待识别图像输入经过训练的深度学习模型,得到由深度学习模型输出的与待识别图像中的每个目标对应的锚框,其中,深度学习模型包括输入模块Input、主干网络Backbone、注意力模块Att、特征融合层Neck和输出模块Head,Att包括通道注意力子模块和空间注意力子模块,Backbone为ShuffleNet v2网络,Head包括3个YOLO Head检测器;对待识别图像中的所有锚框进行计数,得到待识别图像中的目标的数量,如此,有利于减少模型参数,降低运算量。
0/0