-
公开(公告)号：CN118552633A
公开(公告)日：2024-08-27
申请号：CN202410444192.9
申请日：2024-04-12
Applicant: 浪潮电子信息产业股份有限公司
IPC: G06T11/00
Abstract: 本发明提供一种图像生成模型训练方法、图像生成方法、装置及设备，涉及机器学习领域，方法包括：利用训练集对预训练扩散模型进行微调训练，得到第一扩散模型；其中，训练集包含多种类别的训练数据，训练数据包含类别信息和训练图像，扩散模型用于根据输入信息生成图像；将各类别的类别信息输入第一扩散模型，得到第一扩散模型输出的各类别的生成式图像；确定各生成式图像与同一类别的训练图像间的差异度，得到各类别对应的差异度；根据各类别的差异度对训练集中各类别的训练图像数量进行调整，并利用调整后的训练集对第一扩散模型进行微调训练，得到完成训练的第二扩散模型；可提升训练数据稀少的稀有类别在扩散模型中的训练效果。
Search Results
1
records
(took 0.024 seconds)
筛选
AND
AND
过滤
NOT
NOT
Scan to Join
