FocalDEIM

新AI系统RailGen通过生成数据增强铁路入侵检测能力

研究人员开发了RailGen，一个代理引导的图像生成系统，旨在改进铁路轨道上小型异物的检测。该系统利用大型模型创建带有校准入侵的逼真铁路场景，有效地扩充了稀有和小目标的训练数据。同时，还提出了一个互补的检测框架FocalDEIM，以利用这些生成的数据增强训练，提高对小型目标和难例的辨别能力。实验表明，RailGen可以生成像素区域显著减小的物体，并且结合方法在检测精度上优于现有方法。