US 12,406,472 B2
Method of training image generation model, and method of generating image
Donghai Bian, Beijing (CN); Yuwei Wu, Beijing (CN); Guangzhi Sheng, Beijing (CN); Yehan Zheng, Beijing (CN); Weihua Peng, Beijing (CN); and Weijian Xu, Beijing (CN)
Assigned to BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD., Beijing (CN)
Filed by BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD., Beijing (CN)
Filed on Dec. 21, 2022, as Appl. No. 18/086,556.
Claims priority of application No. 202111620055.9 (CN), filed on Dec. 23, 2021.
Prior Publication US 2023/0133981 A1, May 4, 2023
Int. Cl. G06V 10/00 (2022.01); G06V 10/74 (2022.01); G06V 10/771 (2022.01)
CPC G06V 10/771 (2022.01) [G06V 10/761 (2022.01)] 20 Claims
OG exemplary drawing
 
1. A method of training an image generation model, the method comprising:
acquiring a first image sample and a first text sample matched with the first image sample;
performing an enhancement on at least one type of sample in the first image sample and the first text sample according to a predetermined knowledge graph, so as to obtain at least one type of sample in a second image sample obtained by the enhancement and a second text sample obtained by the enhancement; and
training the image generation model according to a training set selected from a first training set, a second training set or a third training set, until the image generation model converges, wherein the first training set comprises the second image sample and the first text sample, the second training set comprises the first image sample and the second text sample, and the third training set comprises the second image sample and the second text sample.