Computer Vision / Video Analytics

Trained on NVIDIA GPUs, Facebook Unveils an AI Fashion Assistant

AI-Generated Summary

  • Fashion++ uses generative adversarial networks to recommend minimal edits that make an outfit more fashionable.
  • The model learns clothing encodings from thousands of publicly available fashionable images trained on NVIDIA P100 and V100 GPUs with cuDNN-accelerated PyTorch.
  • Given an input outfit, the system maps each garment to a code and then updates the encodings to maximize a fashionability score while staying close to the original look.
  • The approach demonstrates how AI can serve as assistive technology by bootstrapping unfashionable examples without resource-intensive human annotation.

Next Steps

Powered by NVIDIA Nemotron. AI-generated content may summarize information incompletely. Verify important information. Learn more

Using GANs, this deep learning-based system can act as a personal fashion designer by recommending changes that can make a person’s outfit more fashionable. 

Using large portions of NVIDIA’s Pix2PixHD code, Facebook AI researchers in collaboration with UT Austin, Cornell University, and Georgia Tech developedFashion++, a deep learning-based model that uses GANs to offer suggestions on what to add, remove, swap, or adjust. 

“Our model consists of a deep image generation neural network that learns to synthesize clothing conditioned on learned per-garment encodings,” the researchers stated in their paper.

At the crux of the work is an activation maximization approach that operates by using encodings from the GAN. the As a first step, Using NVIDIA P100and V100 GPUs, with the cuDNN-accelerated PyTorch deep learning framework, the researchers trained their model on thousands of publicly available images considered to be fashionable.

Once trained, the image-generating neural network identifies and generates the garments or accessories that will best resemble fashionable styles. 

Given an original outfit as the input, the system first maps its composing pieces, which can include items such as a bag, blouse, boots, shirts, to their respective codes. Then, the system uses a discriminative fashionability editing model to gradually update the encodings in the direction that maximizes the outfit’s score, thereby improving its style, the researchers explained. 

According to the researchers, the system will only recommend images closer to the original image or ground truth, helping human users easily implement the proposed changes. 

This work shows the potential of AI as assistive technology. “Our method of bootstrapping unfashionable examples shows how AI systems can learn even without resource-intensive human annotation,” the researchers state. 

A PyTorch implementation of the work is available on GitHub

Read more>

Discuss (0)

Tags