|国家科技期刊平台
首页|期刊导航|哈尔滨工程大学学报|基于深度学习的实例分割边界框回归方法研究

基于深度学习的实例分割边界框回归方法研究OA北大核心CSTPCD

A bounding box regression method of instance segmentation based on deep learning

中文摘要英文摘要

针对实例分割任务中图像中可能出现相互遮挡或边缘模糊导致边界框定位不准确的问题,本文提出了一种新的边界框回归损失函数.将边界框位置预测转化为估计定位置信度随位置变化的概率分布;考虑坐标点间存在联系,提出一种面积差计算方法;为了证明此方法可以很好地应用于先检测后分割的实例分割模型,本文使用Mask R-CNN作为基线.实验结果表明:在边界框检测及实例分割任务中,本文方法的精度优于其他方法,对于小物体的检测与分割效果更显著,训练和评估速度也更快.

A new function of bounding box regression loss in the instance segmentation task is proposed in this paper to solve the problem of inaccurate bounding box location caused by occlusion or blurring of edges in an image.Here,the location prediction of a bounding box is transformed into the probability distribution of the estimated posi-tioning confidence changing with position.A method for area difference calculation is proposed considering the rela-tion between the coordinate points.Then,to prove that the method can be well applied in the instance segmentation model of the detection followed by segmentation,mask R-CNN is taken as the baseline.Experimental results reveal that the proposed method outperforms other methods in terms of accuracy in bounding box detection and instance segmentation.Furthermore,the effect is more significant for small object detection and segmentation,and the train-ing and evaluation speed is faster than others.

刘桂霞;吴彦博;李文辉;王天昊

吉林大学 计算机科学与技术学院,吉林 长春 130012哈尔滨工程大学 船舶工程学院,黑龙江 哈尔滨 150001

计算机与自动化

计算机视觉深度学习卷积神经网络实例分割Mask R-CNN边界框回归KL散度高斯分布

computer visiondeep learningconvolutional neural networkinstance segmentationMask R-CNNbounding box regressionKullback-Leibler divergenceGaussian distribution

《哈尔滨工程大学学报》 2024 (003)

474-479,614 / 7

国家自然科学基金项目(61772226,61862056).

10.11990/jheu.202110049

评论