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

mix ctdet and multi_pose #1001

Open
xudh1991 opened this issue Jul 26, 2022 · 0 comments
Open

mix ctdet and multi_pose #1001

xudh1991 opened this issue Jul 26, 2022 · 0 comments

Comments

@xudh1991
Copy link

I want to integrate detection and key point, I think this method is feasible, My goal is to detect the key points of people when they are detected while detecting the target
for this reason, I changed the annotation of coco key data set, I added other categories except people to the coco key point annotation, and then attached 0 to the key point coordinates and 0 to the number of key points of these categories.
I changed the relevant corresponding code and changed the num_classes to 3
elif opt.task == 'multi_pose': # assert opt.dataset in ['coco_hp'] opt.flip_idx = dataset.flip_idx opt.heads = {'hm': opt.num_classes, 'wh': 2, 'hps': 34} if opt.reg_offset: opt.heads.update({'reg': 2}) if opt.hm_hp: opt.heads.update({'hm_hp': 17}) if opt.reg_hp_offset: opt.heads.update({'hp_offset': 2})
The model results I got:
9multi_pose
My problem is that except for the category of people, other categories cannot obtain accurate results

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

No branches or pull requests

1 participant