diff --git a/docs/0.prepare_dataset.md b/docs/0.prepare_dataset.md index 1dbd62c..5b41914 100644 --- a/docs/0.prepare_dataset.md +++ b/docs/0.prepare_dataset.md @@ -100,5 +100,74 @@ python misc/pretreatment.py --input_path Path_of_OUMVLP-rearranged --output_path ``` +**GREW** + +Step1: Download the data + +Step2: [Unzip](https://github.com/GREW-Benchmark/GREW-Benchmark) the dataset, you will get a structure directory like: + +- Original + ``` + GREW-raw + ├── train + ├── 00001 + ├── 4XPn5Z28 + ├── 00001.png + ├── 00001_2d_pose.txt + ├── 00001_3d_pose.txt + ├── 4XPn5Z28_gei.png + ├── test + ├── gallery + ├── 00001 + ├── 79XJefi8 + ├── 00001.png + ├── 00001_2d_pose.txt + ├── 00001_3d_pose.txt + ├── 79XJefi8_gei.png + ├── probe + ├── 01DdvEHX + ├── 00001.png + ├── 00001_2d_pose.txt + ├── 00001_3d_pose.txt + ├── 01DdvEHX_gei.png + ... + ... + +Step3 : To rearrange directory of OUMVLP dataset, turning to id-type-view structure, Run +``` +python misc/rearrange_GREW.py --input_path Path_of_GREW-raw --output_path Path_of_GREW-rearranged +``` + +Step4: Transforming images to pickle file, run +``` +python misc/pretreatment.py --input_path Path_of_GREW-rearranged --output_path Path_of_GREW-pkl +``` + +- Processed + ``` + GREW-pkl + ├── 00001train (subject in training set) + ├── 00 + ├── 4XPn5Z28 + ├── 4XPn5Z28.pkl + ├──5TXe8svE + ├── 5TXe8svE.pkl + ...... + ├── 00001 (subject in testing set) + ├── 01 + ├── 79XJefi8 + ├── 79XJefi8.pkl + ├── 02 + ├── t16VLaQf + ├── t16VLaQf.pkl + ├── probe + ├── etaGVnWf + ├── etaGVnWf.pkl + ├── eT1EXpgZ + ├── eT1EXpgZ.pkl + ... + ... + ``` + ## Split dataset You can use the partition file in [misc/partitions](misc/partitions/) directly, or you can create yours. Remember to set your path to the partition file in [config/*.yaml](config/). \ No newline at end of file