open-mmlab / mmfashion

Open-source toolbox for visual fashion analysis based on PyTorch
https://open-mmlab.github.io/
Apache License 2.0
1.24k stars 281 forks source link

question about prepare_attr_pred.py #146

Open Danee-wawawa opened 2 years ago

Danee-wawawa commented 2 years ago

Hi, when I use the file prepare_attr_pred.py, one of the parameters is as follows: image

But the dataset directory I downloaded from http://mmlab.ie.cuhk.edu.hk/projects/DeepFashion/AttributePrediction.html is as follows: Attr_Predict ├── Anno_fine │ ├── list_attr_cloth.txt │ ├── train.txt │ ├── ... ├── Anno_coarse │ ├── list_attr_cloth.txt │ ├── ... ├── Eval │ └── list_eval_partition.txt └── Img ├── XXX.jpg └── ...

There is no "Anno" folder. Is this actually the "Anno_fine" folder?

maizer2 commented 2 years ago

Read Prepare an AttrDataset under line


Note that if you use "Anno_fine" that contains 26 better-labeled attributes, nothing else need to be done.

If you use run "Anno_coarse" that contains 1000 roughly labeled attributes, first run the following script to re-organize them. python prepare_attr_pred.py

Please refer to dataset/ATTR_DATASET.md for more info.