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

Key 'center' and 'scale' missing from result list #9

Open
PeteBai opened this issue May 9, 2022 · 3 comments
Open

Key 'center' and 'scale' missing from result list #9

PeteBai opened this issue May 9, 2022 · 3 comments

Comments

@PeteBai
Copy link

PeteBai commented May 9, 2022

image
I used my own dataset for testing purposes. The config file is hrnet_w32_ap10k_256x256. I have downloaded the pre-trained model and removed evaluation code since I don't have ground truth. I have to manually add these 2 keys to the result dict in top_down_transform.py and shared_transform.py to make the code work. Did I miss something?

@PeteBai
Copy link
Author

PeteBai commented May 9, 2022

Also the upper and lower bound of where the prediction region ends also seems to be related to the center coordinates.

@Iven-Wu
Copy link

Iven-Wu commented Jul 18, 2022

The same to me. Have you fixed that problem?

@Iven-Wu
Copy link

Iven-Wu commented Jul 18, 2022

I find the reason. The Config module from mmcv cannot correctly load the dict(type='TopDownGetBboxCenterScale'), in the config file, so the 'scale' and 'center' will disappear. I have no idea how this happens as I look into the source code of mmcv and seems it misses that transform in the beginning.
But one simple way to fix it is to add:
new_transform = {'type': 'TopDownGetBboxCenterScale'} if new_transform not in transforms: transforms.insert(1,new_transform)
in the 'shared_transform.py' right under the 'self.transform = []'

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

2 participants