← 返回列表
一种海报排版模型的训练方法、训练模型和海报排版方法
摘要文本
本发明提供一种海报排版模型的训练方法，包括：获取成品海报数据集、海报前景文字排版数据集；获取预训练隐变量扩散模型的文本‑图像修补模型和预训练光学字符识别模型；基于预训练隐变量扩散模型的文本‑图像修补模型，输入海报前景文字排版数据集，得到海报背景图；将海报前景排版数据集输入预训练光学字符识别模型，得到处理后的海报前景文字排版数据集；建立海报自动排版模型，并基于所述海报背景图、处理后的海报前景排版数据集训练海报自动排版模型，得到完成训练的海报自动排版模型L。本发明还包括对应的训练模型和海报自动排版方法。本发明提供的方案灵活可调、可标注大量海报数据。
申请人信息
- 申请人:北京汉仪创新科技股份有限公司; 上海驿创信息技术有限公司
- 申请人地址:100089 北京市海淀区翠微路2号院五区2号楼二层201
- 发明人: 北京汉仪创新科技股份有限公司; 上海驿创信息技术有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种海报排版模型的训练方法、训练模型和海报排版方法 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311551424.2 |
| 申请日 | 2023/11/21 |
| 公告号 | CN117332839A |
| 公开日 | 2024/1/2 |
| IPC主分类号 | G06N3/08 |
| 权利人 | 北京汉仪创新科技股份有限公司; 上海驿创信息技术有限公司 |
| 发明人 | 章伟希 |
| 地址 | 北京市海淀区翠微路2号院五区2号楼二层201; 上海市虹口区广纪路838号B幢616 |
专利主权项内容
1.一种海报排版模型的训练方法，其特征在于，包括以下步骤：步骤S1、获取成品海报数据集、海报前景文字排版数据集；步骤S2、获取预训练隐变量扩散模型的文本-图像修补模型和预训练光学字符识别模型；步骤S3、基于所述预训练隐变量扩散模型的文本-图像修补模型，输入所述海报前景文字排版数据集，得到海报背景图；步骤S4、将所述海报前景排版数据集输入所述预训练光学字符识别模型，得到处理后的海报前景文字排版数据集；步骤S5、建立海报自动排版模型L1，并基于所述海报背景图、所述处理后的海报前景排版数据集训练海报自动排版模型L1，得到完成训练的海报自动排版模型L。