AI-driven design platform

A non-transitory computer-readable storage medium coupled to one or more processors and having instructions stored thereon which, when executed by the one or more processors, cause the one or more processors to perform operations for generating a design using an artificial intelligence (AI)-based platform comprising a suite of AI-based design assistants, the operations comprising:

• receiving, by a design generation assistant of the AI-based design platform, a content image depicting a first set of objects and a style image depicting a second set of objects and a collective style;
• processing, by the design generation assistant, the content image using a set of machine learning (ML) models to determine a first sub-set of objects comprising one or more objects, to which a style is to be transferred from the style image, the first sub-set of objects comprising fewer objects than the first set of objects, the set of ML models comprising a first ML model to identify objects in the content image and a second ML model and a third ML model to identify planar surfaces in the content image to segment the objects and determine the first sub-set of objects;
• processing, by the design generation assistant, the style image using the set of ML models to determine a second sub-set of objects comprising one or more objects, from which the style is to be transferred to objects in the first sub-set of objects;
• and
• generating, by the design generation assistant, an output image depicting one or more stylized objects, each stylized object comprising an object in the first sub-set of objects having the style of the one or more objects in the second sub-set of objects, a remainder of the output image comprising un-stylized content of the content image.

A system, comprising:

• one or more computers;
• and
• a computer-readable storage device coupled to the computing device and having instructions stored thereon which, when executed by the computing device, cause the computing device to perform operations for generating a design using an artificial intelligence (AI)-based platform comprising a suite of AI-based design assistants, the operations comprising:
• receiving, by a design generation assistant of the AI-based design platform, a content image depicting a first set of objects and a style image depicting a second set of objects and a collective style;
• processing, by the design generation assistant, the content image using a set of machine learning (ML) models to determine a first sub-set of objects comprising one or more objects, to which a style is to be transferred from the style image, the first sub-set of objects comprising fewer objects than the first set of objects, the set of ML models comprising a first ML model to identify objects in the content image and a second ML model and a third ML model to identify planar surfaces in the content image to segment the objects and determine the first sub-set of objects;
• processing, by the design generation assistant, the style image using the set of ML models to determine a second sub-set of objects comprising one or more objects, from which the style is to be transferred to objects in the first sub-set of objects;
• and
• generating, by the design generation assistant, an output image depicting one or more stylized objects, each stylized object comprising an object in the first sub-set of objects having the style of the one or more objects in the second sub-set of objects, a remainder of the output image comprising un-stylized content of the content image.