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
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
...
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/
.