anibali / h36m-fetch

Human 3.6M 3D human pose dataset fetcher
Apache License 2.0
367 stars 48 forks source link

Processing all frames in process_all.py #28

Closed jamalknight closed 4 years ago

jamalknight commented 4 years ago

The process_all.py script processes every 64th frame for the subjects 'S9' and 'S11' in lines:

  if subject == 'S9' or subject == 'S11':
        return np.arange(0, len(poses_3d_univ), 64)

How can I change the above code to process all frames, without skipping any?

Tried: Replacing 64 with 1 and all but did not work

anibali commented 4 years ago

It should just be a matter of replacing

https://github.com/anibali/h36m-fetch/blob/b8112ff5525aa4ddddabd4c7f629df6752edad20/process_all.py#L39

with

        return np.arange(0, len(poses_3d_univ))
jamalknight commented 4 years ago

great, thanks