[发明专利]一种铁路货车装载状态图像切割方法、装置、设备及介质

说明书

本发明涉及一种铁路货车装载状态图像切割方法、装置、设备及介质，该方法包括：获取待处理铁路货车的至少两张图像；对所述至少两张图像进行横向拼接，得到待处理图像；采用设定大小的窗口对所述待处理图像进行裁剪，得到多张裁剪图像；根据多张所述裁剪图像和预训练的定位模型，确定所述待处理铁路货车的连接处位置；根据所述连接处位置，对所述待处理图像进行切割，得到包含单节车厢的图像。通过本发明的方法，可以解决因为外在环境变化而影响切割质量的问题，进一步提升了铁路货运视频监控人员的看图质量与效率。

技术领域

本发明涉及铁路货运安全检测技术领域，具体而言，本发明涉及一种铁路货车装载状态图像切割方法、装置、设备及介质。

背景技术

目前铁路货车装载状态图像切割主要使用红外对射，当车列通过红外发射装置、遮挡红外射线横向穿过车体时，红外射线接收装置不生效，被判定为一节货车车辆，直至铁路货车行驶至货车车辆连接处，接收在货车车辆连接处的信号并根据该信号切割图像。但此种方式是使用物理设备(红外对射设备)切割图像的方式，容易受到环境因素的影响，例如，轨面在列车通过时的震动，容易使得红外对射设备发生偏移，导致射线发送与接收装置间无法对准进而造成的车辆切割错误，极大的影响铁路货运视频监控人员的看图质量与效率，进一步影响图像智能识别质量与效率。

发明内容

本发明所要解决的技术问题是提供了一种铁路货车装载状态图像切割方法、装置、设备及介质，旨在解决上述至少一个技术问题。

本发明解决上述技术问题的技术方案如下：一种铁路货车装载状态图像切割方法，该方法包括：

获取待处理铁路货车的至少两张图像；

对所述至少两张图像进行横向拼接，得到待处理图像；

采用设定大小的窗口对所述待处理图像进行裁剪，得到多张裁剪图像；

根据多张所述裁剪图像和预训练的定位模型，确定所述待处理铁路货车的连接处位置；

根据所述连接处位置，对所述待处理图像进行切割，得到包含单节车厢的图像。

本发明的有益效果是：本申请方案中，不借助于红外对射装置进行图像切割，而是采用预先训练好的定位模型先定位出待处理图像中的连接处位置，再根据连接处位置进行切割，可以解决因为外在环境变化而影响切割质量的问题，进一步提升了铁路货运视频监控人员的看图质量与效率。

在上述技术方案的基础上，本发明还可以做如下改进。

进一步，若上述待处理图像小于设定尺寸，所述设定尺寸为所述定位模型的输入图像的尺寸，所述方法还包括：

将所述待处理图像进行扩大处理，得到处理后的图像；

所述采用设定大小的窗口对所述待处理图像进行裁剪，得到多张裁剪图像，包括：

采用设定大小的窗口对所述处理后的图像进行裁剪，得到多张裁剪图像。

采用上述进一步方案的有益效果是，模型对输入的图像的尺寸通常有要求，则对于不满足该要求的图像，即图像尺寸小于设定尺寸的待处理图像，可先对待处理图像进行尺寸扩大，再进行后续的处理。

进一步，若上述连接处位置包括至少两个连接处位置，所述根据所述连接处位置，对所述待处理图像进行切割，得到包含单节车厢的图像，包括：

对于每两个所述连接处位置，根据两个所述连接处位置，对所述待处理图像进行切割，得到两个所述连接处位置对应的包含单节车厢的图像，所述包含单节车厢的图像为两个所述连接处位置之间的像素所对应的图像，且两个所述连接处位置之间的像素距离大于设定值。