基于R2Net的水下鱼类目标检测方法、装置及存储介质

摘要

本发明提出一种基于R2Net的水下鱼类目标检测方法、装置及存储介质，涉及目标检测技术领域。在进行水下鱼类目标检测时，首先获取训练用水下鱼类图像；将所述训练用水下鱼类图像输入R2Net模型的复杂网络结构进行训练，得到R2Net模型的复杂网络结构的网络参数；根据R2Net模型的复杂网络结构的网络参数，对R2Net模型的复杂网络结构进行等效变换，得到R2Net模型的简化网络结构；R2Net模型的简化网络结构为Kmax×Kmax等效卷积核；获取待检测水下鱼类图像；将所述待检测水下鱼类图像输入R2Net模型的简化网络结构，得到特征提取结果；基于特征提取结果进行水下鱼类目标检测，得到水下鱼类目标检测结果。本发明利用R2Net模型进行水下鱼类目标检测，实现了高精度的快速水下鱼类目标检测。