TotalCapture-Toolbox

TotalCapture-Toolbox copied to clipboard

Dependencies

We use python 3 for this project.

Make sure FFMPEG is installed, or install by sudo apt install ffmpeg

Install the python packages dependencies by pip install -r requirements.txt

Preparing Data

Download the data from Total Capture official site.

Please Note that we have NO permission to redistribute this dataset. Please never ask us for a copy.

Then organize the data in below structure (put all data in ./data/totalcapture):

./data/totalcapture
├── calibration.cal
├── s1
│   ├── acting1
│   │   ├── acting1_BlenderZXY_YmZ.bvh
│   │   ├── acting1_Xsens_AuxFields.sensors
│   │   ├── gt_skel_gbl_ori.txt
│   │   ├── gt_skel_gbl_pos.txt
│   │   ├── s1_acting1_calib_imu_bone.txt
│   │   ├── s1_acting1_calib_imu_ref.txt
│   │   ├── s1_acting1_Xsens.sensors
│   │   ├── TC_S1_acting1_cam1.mp4
│   │   ├── TC_S1_acting1_cam2.mp4
│   │   ├── TC_S1_acting1_cam3.mp4
│   │   ├── TC_S1_acting1_cam4.mp4
│   │   ├── TC_S1_acting1_cam5.mp4
│   │   ├── TC_S1_acting1_cam6.mp4
│   │   ├── TC_S1_acting1_cam7.mp4
│   │   └── TC_S1_acting1_cam8.mp4
│   ├── acting2
│   │   ├── ...
│   ├── ...
├── s2
...

Usage

python gendata/gendb.py

# if cannot import tools
# export PYTHONPATH=".:$PYTHONPATH"; python gendata/gendb.py  

Options: You can change options in gendata/config.yaml, e.g.

• save_frame: whether extract frames from the videos
• gen_train: generate dataset for training
• gen_test: generate dataset for testing

Finally, you should have a bunch of images and two .pkl files in the ./data/images. Move the .pkl files to ../annot directory, and you get

./data/
├── images
│   ├── s_01_act_01_subact_01_ca_01
│   │   ├── 000000.jpg
│   │   ├── ...
│   ├── ...
├── annot
│   ├── totalcapture_train.pkl
│   ├── totalcapture_validation.pkl

If you want to archive all images in one zip file which is more efficient to be transferred to server or cloud storage, make sure your pwd is parent directory of images/, then run zip -0 -r images.zip images/.