Mmfashion
Open-source toolbox for visual fashion analysis based on PyTorch
Install / Use
/learn @open-mmlab/MmfashionREADME
MMFashion
<p align="center"> <img src='./misc/logo_mmfashion.png' width=320> </p>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.
[2020-12-08] MMFashion v0.5 is released, adding virtual try-on 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
- [x] Fashion Attribute Prediction
- [x] Fashion Recognition and Retrieval
- [x] Fashion Landmark Detection
- [x] Fashion Parsing and Segmentation
- [x] Fashion Compatibility and Recommendation
- [x] 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
- Xin Liu (veralauee)
- Jiancheng Li (lijiancheng0614)
- Jiaqi Wang (myownskyW7)
- Ziwei Liu (liuziwei7)
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}
}
