[发明专利]数据集采集方法和装置、存储介质和电子设备

说明书

本公开涉及一种数据集采集方法和装置、存储介质和电子设备，属于深度学习领域，能够将广角镜头采集到的图像数据集直接应用到目标检测相关算法中。该数据集采集方法，该方法包括：接收物体位于广角镜头的第i个子视角范围内时的图像，其中所述广角镜头的整个视角范围被划分成了N个子视角范围，1≤i≤N；以及基于所述广角镜头的畸变特点，利用所接收的第i个子视角范围内的图像来扩充所述物体在所述广角镜头的其余子视角范围内的图像。

技术领域

本公开涉及深度学习领域，具体地，涉及一种数据集采集方法和装置、存储介质和电子设备。

背景技术

广角镜头的视角大、焦距短，导致其采集到的图像中的物体会发生畸变，进而导致同一物体的本身特征无法被各种图像处理算法直接利用。目前处理这种畸变的主流方法是先将畸变的图像进行畸变校正。然而，目前广角镜头的畸变校正算法的校正效果并不好，而且校正后的图像仅适用于人眼观察，还无法将校正后的图片应用到目标检测相关算法中。

发明内容

本公开的目的是提供一种数据集采集方法和装置、存储介质和电子设备，能够将广角镜头采集到的图像数据集直接应用到目标检测相关算法中。

根据本公开的第一实施例，提供一种数据集采集方法，该方法包括：接收物体位于广角镜头的第i个子视角范围内时的图像，其中所述广角镜头的整个视角范围被划分成了N个子视角范围，1≤i≤N；以及基于所述广角镜头的畸变特点，利用所接收的第i个子视角范围内的图像来扩充所述物体在所述广角镜头的其余子视角范围内的图像。

可选地，所述接收物体位于广角镜头的第i个子视角范围内时的图像，包括：接收所述物体在所述第i个子视角范围内的至少一个固定位置上每原地旋转预设角度时的图像，以得到所述物体的360度特征信息。

可选地，所述物体在所述第i个子视角范围内的至少一个固定位置上的原地旋转的次数M为：M＝360°/所述预设角度。

可选地，所述基于所述广角镜头的畸变特点，利用所接收的第i个子视角范围内的图像来扩充所述物体在所述广角镜头的其余子视角范围内的图像，包括：利用所述广角镜头的桶形畸变特点，将所接收的第i个子视角范围内的图像以所述广角镜头的整个视角范围的中心为中心点进行旋转，来扩充所述物体在所述广角镜头的其余子视角范围内的图像。

可选地，所接收的第i个子视角范围内的图像以所述广角镜头的整个视角范围的中心为中心点进行旋转的旋转角度r为：r＝360°/N。

根据本公开的第二实施例，提供一种数据集采集装置，该装置包括：接收模块，用于接收物体位于广角镜头的第i个子视角范围内时的图像，其中所述广角镜头的整个视角范围被划分成了N个子视角范围，1≤i≤N；以及扩充模块，用于基于所述广角镜头的畸变特点，利用所述接收模块接收到的第i个子视角范围内的图像来扩充所述物体在所述广角镜头的其余子视角范围内的图像。

可选地，所述接收模块包括接收子模块，用于接收所述物体在所述第i个子视角范围内的至少一个固定位置上每原地旋转预设角度时的图像，以得到所述物体的360度特征信息。

可选地，所述物体在所述第i个子视角范围内的至少一个固定位置上的原地旋转的次数M为：M＝360°/所述预设角度。

可选地，所述扩充模块包括扩充子模块，用于利用所述广角镜头的桶形畸变特点，将所述接收模块所接收的第i个子视角范围内的图像以所述广角镜头的整个视角范围的中心为中心点进行旋转，来扩充所述物体在所述广角镜头的其余子视角范围内的图像。

可选地，所述接收模块所接收的第i个子视角范围内的图像以所述广角镜头的整个视角范围的中心为中心点进行旋转的旋转角度r为：r＝360°/N。

根据本公开的第三实施例，提供一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现根据本公开第一实施例所述方法的步骤。