Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Custom Dataset Training with PoseC3D problem #1293

Closed
Disisxhy opened this issue Nov 26, 2021 · 5 comments · Fixed by #1295
Closed

Custom Dataset Training with PoseC3D problem #1293

Disisxhy opened this issue Nov 26, 2021 · 5 comments · Fixed by #1295

Comments

@Disisxhy
Copy link

Once I skip the 'ntu_det_postproc' step as noticed and then run 'python ntu_pose_extraction.py S001C001P001R001A001_rgb.avi S001C001P001R001A001_rgb.pkl', it shows the error below:

image
image

Hope somebody can help

@kennymckormick
Copy link
Member

Hello, in PR #1295, I add an option named 'skip_postproc'. Please try to extract pose with this flag, and check if it can resolve your issues.

@Disisxhy
Copy link
Author

the ‘skip_postproc' doesn't work well. comes the error 'AttributeError: 'abstractproperty' object has no attribute 'skip_postproc'

since the 'skip' is not necessary for me, I simply change ' num_frame, num_person ' thing and it woks perfectly! thx so much!

@bit-scientist
Copy link
Contributor

Hi, @Disisxhy, basically you don't exactly have to use ntu_pose_extraction.py as it is a burden to switch off ntu_pose_extraction function, you can modify demo_posec3d.py a bit to extract necessary keys from fake_anno (in demo_posec3d.py) and dump into a pickle file. Note that you may have to use your video's name for frame_dirs value and, its string split for labels value (if the video name contains class label ) as shown at #1258. Hope that helps.

@shelly-xue
Copy link

@bit-scientist @kennymckormick hi ,my custom videos have two people, if i use ntu_pose_extraction.py to extract poses, Can it be used directly for training? because we need to maintain the continuity of everyone's time, am i right?

@kennymckormick
Copy link
Member

I think it can be directly used for PoseC3D training, but maybe not ST-GCN. Since for ST-GCN, the skeleton input should be several continuous sequences, while ntu_pose_extraction.py does not guarantee this: the person 1 in frame t and the person 1 in frame t + 1 may not be the same person.

@bit-scientist @kennymckormick hi ,my custom videos have two people, if i use ntu_pose_extraction.py to extract poses, Can it be used directly for training? because we need to maintain the continuity of everyone's time, am i right?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

4 participants