一种用于目标检测任务的图像数据增强方法

本发明属于图像处理技术领域，公开一种用于目标检测任务的图像数据增强方法。通过随机生成模块随机生成不同几何形状的X射线违禁品图像；采集X射线伪彩色图像和双能X射线RAW图像，X射线伪彩色图像作为输入值，双能X射线RAW图像作为真值，共同训练图像转换网络；X射线违禁品图像经训练后的图像转换网络生成双能X射线违禁品RAW图像，X射线包裹图像经训练后的图像转换网络生成双能X射线包裹RAW图像；通过双能X射线成像原理设计数学模型，所述数学模型建立双能X射线违禁品RAW图像和双能X射线包裹RAW图像的成像关系，从而合成X射线违禁品包裹RAW图像，并进行伪彩色成像，获得X射线违禁品包裹图像。