open-mmlab / mmfashion

Open-source toolbox for visual fashion analysis based on PyTorch
https://open-mmlab.github.io/
Apache License 2.0
1.27k stars 283 forks source link
attribute-prediction clothes-retrieval deep-learning fashion-ai landmark-detection visual-fashion-analysis

MMFashion

Introduction

[Technical Report]

MMFashion is an open source visual fashion analysis toolbox based on PyTorch. It is a part of the open-mmlab project developed by Multimedia Lab, CUHK.

Updates

[2019-11-01] MMFashion v0.1 is released.

[2020-02-14] MMFashion v0.2 is released, adding consumer-to-shop retrieval module.

[2020-04-27] MMFashion v0.3 is released, adding fashion segmentation and parsing module.

[2020-05-04] MMFashion v0.4 is released, adding fashion compatibility and recommendation module.

[2020-12-08] MMFashion v0.5 is released, adding virtual try-on module.

Features

Requirements

Installation

git clone --recursive https://github.com/open-mmlab/mmfashion.git
cd mmfashion
python setup.py install

Another option: Docker Image

We provide a Dockerfile to build an image.

# build an image with PyTorch 1.5, CUDA 10.1
docker build -t mmfashion docker/

Run it with

docker run --gpus all --shm-size=8g -it mmfashion

Get Started

Please refer to GETTING_STARTED.md for the basic usage of MMFashion.

Data Preparation

Please refer to DATA_PREPARATION.md for the dataset specifics of MMFashion.

Model Zoo

Please refer to MODEL_ZOO.md for a comprehensive set of pre-trained models in MMFashion.

Contributing

We appreciate all contributions to improve MMFashion. Please refer to CONTRIBUTING.md for the contributing guideline.

Related Tools

License

This project is released under the Apache 2.0 license.

Team

Citation

@inproceedings{mmfashion,
  title={MMFashion: An Open-Source Toolbox for Visual Fashion Analysis},
  author={Liu, Xin and Li, Jiancheng and Wang, Jiaqi and Liu, Ziwei},
  booktitle={{ACM Multimedia 2021, Open Source Software Competition},
  year={2021}
}