基于生成模型的安全帽数据生成方法及装置

本发明涉及基于生成模型的安全帽数据生成方法及装置，该方法包括：获取安全帽数据，标记文本提示对话框和安全帽边界框，对安全帽数据进行分割标注。通过分割标注后的安全帽数据对Mobile SAM轻量化分割模型进行训练和微调，通过文本提示对话框对CLIP编码器进行微调。调用训练好的Mobile SAM轻量化分割模型对安全帽边界框内的安全帽数据进行分割处理，得到安全帽的掩码信息。基于文本提示对话框、安全帽边界框以及掩码信息，对Gligen模型进行微调和约束，得到用于生成安全帽数据的Gligen模型。将当前安全帽图像作为Gligen模型的输入，调用Gligen模型输出当前安全帽图像对应的安全帽数据。