一种基于复杂模型的压缩剪枝的船舶检测方法及系统
摘要:
本发明公开一种基于复杂模型的压缩剪枝的船舶检测方法及系统。其中,该方法通过修改卷积核尺寸、卷积通道数量、主干网络的每个模块的卷积核个数,有效提高了原始船舶检测模型的复杂度和感受野,从而提高了船舶检测的精度;通过将层卷积、卷积通道和卷积核的较低权重置为0,减少了模型的参数量,从而降低了模型对硬件算力的需求,加快了船舶检测速度;采用分阶段模型训练,对模型修改时进行微调,从而实现在保证船舶检出率的同时,加快船舶模型的推理速度;将权重为0的卷积通道和卷积核进行删除,从而降低模型对硬件算力需求,进一步加快了船舶检测速度。
0/0