/Mannequin-Data-Download-and-Extract

Downloading and Extracting the Images of Mannequin paper Dataset

Primary LanguagePython

Mannequin-Data-Download-and-Extract

This is a python implementation of downloading and extracting the videos for mannequin paper "Learning the Depths of Moving People by Watching Frozen People."

download txt files

  1. download the mannequin dataset from here
  2. extract the files in a directory (e.g. ~/MannequinChallenge)

download videos

  1. I use pytube3 for downloading the videos from youtube. more information at here

  2. Install pytube3 for python3:

pip install pytube3
  1. make a folder code in the extracted directory each folder (train,test, and validation) and put the download.py there (e.g. ~/MannequinChallenge/train/code/download.py)

  2. run the download.py:

python download.py
  1. your data will be downloaded in data folder in extracted directory (e.g. ~/MannequinChallenge/train/data/) each folder in the data directory represent each txt file.

extract the frames

  1. once you downloaded the videos from youtube, you should extract the frames for each video based on the frame timesteps given in the txt files.

  2. for that I use opencv library. you can install it:

pip install opencv-python
  1. again for each dataset (train,test, or validation) put the extract.py in the code folder (e.g. ~/MannequinChallenge/train/code/extract.py)

  2. run extract.py:

python extract.py