OpenGait/docs/0.prepare_dataset.md

Prepare dataset

Suppose you have downloaded the original dataset, we need to preprocess the data and save it as pickle file. Remember to set your path to the root of processed dataset in config/*.yaml.

Preprocess

CASIA-B

Download URL: http://www.cbsr.ia.ac.cn/GaitDatasetB-silh.zip

• Original

  CASIA-B
      001 (subject)
          bg-01 (type)
              000 (view)
                  001-bg-01-000-001.png (frame)
                  001-bg-01-000-002.png (frame)
                  ......
              ......
          ......
      ......
  

• Run python misc/pretreatment.py --input_path CASIA-B --output_path CASIA-B-pkl
• Processed

  CASIA-B-pkl
      001 (subject)
          bg-01 (type)
                  000 (view)
                      000.pkl (contains all frames)
              ......
          ......
      ......
  

OUMVLP

Step1: Download URL: http://www.am.sanken.osaka-u.ac.jp/BiometricDB/GaitMVLP.html

Step2: Unzip the dataset, you will get a structure directory like:

python misc/extractor.py --input_path Path_of_OUMVLP-base --output_path Path_of_OUMVLP-raw --password Given_Password

• Original

  OUMVLP-raw
      Silhouette_000-00 (view-sequence)
          00001 (subject)
              0001.png (frame)
              0002.png (frame)
              ......
          00002
              0001.png (frame)
              0002.png (frame)
              ......
          ......
      Silhouette_000-01
          00001
              0001.png (frame)
              0002.png (frame)
              ......
          00002
              0001.png (frame)
              0002.png (frame)
              ......
          ......
      Silhouette_015-00
          ......
      Silhouette_015-01
          ......
      ......
  

Step3 : To rearrange directory of OUMVLP dataset, turning to id-type-view structure, Run

python misc/rearrange_OUMVLP.py --input_path Path_of_OUMVLP-raw --output_path Path_of_OUMVLP-rearranged

Step4: Transforming images to pickle file, run

python misc/pretreatment.py --input_path Path_of_OUMVLP-rearranged --output_path Path_of_OUMVLP-pkl

• Processed

  OUMVLP-pkl
      00001 (subject)
          00 (sequence)
              000 (view)
                  000.pkl (contains all frames)
              015 (view)
                  015.pkl (contains all frames)
              ...
          01 (sequence)
              000 (view)
                  000.pkl (contains all frames)
              015 (view)
                  015.pkl (contains all frames)
              ......
      00002 (subject)
          ......
      ......
  

Split dataset

You can use the partition file in misc/partitions directly, or you can create yours. Remember to set your path to the partition file in config/*.yaml.