基于模型轻量化的目标检测方法
    3.
    发明公开

    公开(公告)号:CN116229199A

    公开(公告)日:2023-06-06

    申请号:CN202211612259.2

    申请日:2022-12-15

    Abstract: 一种基于模型轻量化的目标检测方法,步骤包括:首先采集图片;再把图片送入目标检测方法模型进行检测;最后得到图片中的目标信息。目标检测方法模型的构建步骤包括:步骤1:处理训练集,将图片数据集格式转化为适合YOLOv5训练的xml格式,并将之划分为训练集和测试集两大部分;步骤2:搭建PyTorch深度学习框架,其中深度学习模型的配置利用YOLOv5算法,步骤包括:步骤3:修改YOLOv5网络;步骤4:训练网络;步骤5:利用训练好的模型进行探测detect操作,从而得到基于模型轻量化的目标检测方法模型。本方法通过多种措施修改YOLOv5网络,使模型轻量化,适于嵌入到芯片上。

Patent Agency Ranking