Skip to content
New issue

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

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

Already on GitHub? # to your account

[Feature] Add RTMW3D for 3D wholebody pose estimation task #3037

Merged
merged 18 commits into from
Jul 12, 2024

Conversation

xiexinch
Copy link
Collaborator

@xiexinch xiexinch commented May 6, 2024

Motivation

Add RTMPose3D model.

Modification

  • Add rtmpose3d to projects
  • Minor change to data pipeline and 3d datasets

BC-breaking (Optional)

Use cases (Optional)

Checklist

Before PR:

  • I have read and followed the workflow indicated in the CONTRIBUTING.md to create this PR.
  • Pre-commit or linting tools indicated in CONTRIBUTING.md are used to fix the potential lint issues.
  • Bug fixes are covered by unit tests, the case that causes the bug should be added in the unit tests.
  • New functionalities are covered by complete unit tests. If not, please add more unit tests to ensure correctness.
  • The documentation has been modified accordingly, including docstring or example tutorials.

After PR:

  • CLA has been signed and all committers have signed the CLA in this PR.

Comment on lines 182 to 186
img_path = f'{self.data_root}original/{subject}/Images/{act}.{cam}/frame_{frames[frame_ids[-1]]}.jpg' # noqa
img_paths = [
f'{self.data_root}original/{subject}/Images/{act}.{cam}/frame_{frames[i]}.jpg' # noqa
for i in frame_ids
]
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I suggest using osp.join() to improve robustness.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

In case there is no '/' in self.data_root

@xiexinch xiexinch changed the title [Feature] Add RTMPose3D-Wholebody [Feature] Add RTMW3D for 3D wholebody pose estimation task Jul 12, 2024
@xiexinch xiexinch merged commit 08bf67b into open-mmlab:dev-1.x Jul 12, 2024
5 of 15 checks passed
@stihuangyuan
Copy link

rtmpose3d train dataset, h3wb_dataset h3wb_train_bbox.npz ann file could not find!

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

Successfully merging this pull request may close these issues.

4 participants