itsuncheng / DeepStyle

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DeepStyle: Using State-of-the-Art Deep Learning to Generate Realistic High Fashion Clothing and Styles

This is the official repository for the Medium Article Part 1 and Part 2 that contains all the relevant source code, jupyter notebooks, and configuration files for the paper. The paper is also included in the repository and can be found here.

Brief Description of the Files

The .yaml files contain the configuration details of the object detection models trained in this research as described in the paper.

The DeepFashion2Coco.ipynb shows the steps to convert the DeepFashion Dataset to COCO format to ease the training and validation of the models implemented by the Detectron framework.

The vis.py file is under the detectron/utils/ folder in the Detectron github repository. Replace the vis.py file in the repository with the updated one to reproduce the cropping of the test images based on the predicted bounding boxes, as described in the paper. This vis.py adds the code for detecting full-body fashion items in images, cropping them based on the bounding boxes, and saving the results to a local directory.

Finally, the rest of the jupyter notebooks provide the implementation details of the DCGANs as described in paper. The hyperparameters for training and the architectures for the generator and discriminator are provided for each model tested. Code are written in Pytorch.

About


Languages

Language:Jupyter Notebook 100.0%Language:Python 0.0%