crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

GREW dataset pretreatment

Browse Source
This commit is contained in:
Noah
2022-03-21 19:26:21 +08:00
parent 84f58f60aa
commit 4c8d272eff
1 changed files with 69 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/0.prepare_dataset.md
+69
View File
@@ -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/).