一种火焰数据生成方法及终端
摘要:
本发明公开一种火焰数据生成方法及终端,获取训练样本集和预设背景图像;使用所述训练样本集对火焰生成网络进行训练,得到训练后的火焰生成网络;使用所述训练后的火焰生成网络对所述预设背景图像进行随机火焰生成,得到带有火焰的预设背景图像;将所述带有火焰的预设背景图像输入预设火焰分割网络进行火焰分割,得到火焰图像,并将所述火焰图像与所述预设背景图像进行合成,生成火焰数据,能够对特定的预设背景图像生成随机火焰,形成原图某个位置起火的效果,使得最终的火焰数据更加符合实际情况,更加真实、有效,从而有效增加了火焰检测的样本数据,解决了数据样本量不足以及火焰数据缺乏多样性的问题。
公开/授权文献
0/0