The official repository of "FashionFail: Addressing Failure Cases in Fashion Object Detection and Segmentation".
Create a new environment named ff
(short for FashionFail):
conda create -n ff python=3.9
conda activate ff
Then, clone the repository and install the package via:
git clone https://github.com/rizavelioglu/fashionfail.git
cd fashionfail
pip install -e .[dev] # full development, e.g. training, inference, evaluation
Note: If you just want to construct the FashionFail dataset, you could install the package via the following instead, which avoids downloading heavy packages:
pip install -e . # basic usage, i.e. downloading dataset
We offer a comprehensive explanation of each step, including dataset creation, training, and evaluation. You can find detailed documentation at references/README.md
The dataset (annotations and image URLs) is available at .
Execute the following to download the dataset (which downloads the annotation files and images for each train
, val
,
and test
splits):
python fashionfail/data/make_dataset.py \
--save_dir "dir/to/save" \ # [optional] default: "~/.cache/fashionfail/"
An optional argument --save_dir
can be set to construct the dataset in the preferred directory, but it is not
recommended to alter the default location, as the training script expects the data to be at a specific location.
Trained models are available at . To run inference using models execute the following, which downloads the model if necessary:
python fashionfail/models/predict_models.py \
--model_name "facere" \ # or "facere+"
--out_dir "<OUTPUT_DIR>" \ # predictions are stored here
--image_dir "<IMAGES_DIR>" # image dir to run inference for
Alternatively, you can check out the demo in a browser at and run inference using the user interface.
Please refer detailed explanation of training both
Facere
andFacere+
models at references/03_training.md
And see the inference for
Attribute Mask R-CNN
andFashionformer
at references/04_inference.md
The following project/directory structure is adopted: Cookiecutter Data Science-v1 by DrivenData.
.
├── LICENSE
├── notebooks/ # Jupyter Notebooks
├── pyproject.toml # The requirements file for reproducing the environment
├── README.md
├── references/ # Explanatory materials
├── src/
│ └── data/ # Scripts to download or construct dataset
│ └── features/ # Scripts to turn raw data into features
│ └── models/ # Scripts to train models, use trained models to make predictions
│ └── visualization/ # Scripts to create exploratory and results oriented visualizations
TL;DR: Not available for commercial use, unless the FULL source code is open-sourced!
This project is intended solely for academic research. No commercial benefits are derived from it.
The code, datasets, and models are published under the Server Side Public License (SSPL).
If you find this repository useful in your research, please consider giving a star ⭐ and a citation:
@inproceedings{velioglu2024fashionfail,
author = {Velioglu, Riza and Chan, Robin and Hammer, Barbara},
title = {FashionFail: Addressing Failure Cases in Fashion Object Detection and Segmentation},
booktitle = {IJCNN},
year = {2024},
doi = {https://doi.org/ng59},
eprint = {2404.08582}
}