/mmfashion

Open-source toolbox for visual fashion analysis based on PyTorch

Primary LanguagePythonApache License 2.0Apache-2.0

MMFashion

Introduction

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.

Features

  • Flexible: modular design and easy to extend

  • Friendly: off-the-shelf models for layman users

  • Comprehensive: support a wide spectrum of fashion analysis tasks

    • Fashion Attribute Prediction
    • Fashion Recognition and Retrieval
    • Fashion Landmark Detection
    • Fashion Parsing and Segmentation
    • Fashion Compatibility and Recommendation
    • Fashion Virtual Try-On

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

@article{mmfashion,
  title={MMFashion: An Open-Source Toolbox for Visual Fashion Analysis},
  author={Liu, Xin and Li, Jiancheng and Wang, Jiaqi and Liu, Ziwei},
  journal= {arXiv preprint arXiv:2005.08847},
  year={2020}
}