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

--weights_path ./weight/X-152.pth where download? #2

Open
alice-cool opened this issue Sep 28, 2022 · 10 comments
Open

--weights_path ./weight/X-152.pth where download? #2

alice-cool opened this issue Sep 28, 2022 · 10 comments
Labels
enhancement New feature or request help wanted Extra attention is needed

Comments

@alice-cool
Copy link

Dear scholar, I want to extract the 16 times 16 features using the ResNext152 described in your paper.

@alice-cool
Copy link
Author

@alice-cool
Copy link
Author

alice-cool commented Sep 28, 2022

When I INPUT "CUDA_VISIBLE_DEVICES=1 python extract_trar_grid_feature.py --config-file configs/X-152-grid.yaml --dataset coco_2015_test --weight_path ./weight/X-152.pth --output_dir ./data/" to extract from coco test2015.zip jpg.
It throws the exception like the following:

**raise KeyError(
KeyError: "**Dataset 'coco_2015_test' is not registered!

image
When adding the code to fix it and entering a new window again, the error eliminates.
image

But it didn't give any information about the number of instances over 80 categories.

@alice-cool
Copy link
Author

alice-cool commented Sep 29, 2022

image
When using multiple GPUs to run the code, the code throws the exception like the above picture...

Now I did the following modifications:
①in train_egine.py
image

②in test_egine.py
Just narrow tab it.
image

And the error eliminate. Because of the large scale input feature, the code is running for a long time....

@rentainhe
Copy link
Owner

https://github.com/facebookresearch/grid-feats-vqa is it right?

Yes, you can use this code to extract 16x16 features, and we have also released our feature-extraction code here: TRAR-Feature-Extraction, maybe you can extract 16x16 features by modifing the configs

@rentainhe
Copy link
Owner

rentainhe commented Sep 29, 2022

Dear scholar, I want to extract the 16 times 16 features using the ResNext152 described in your paper.

Maybe you can download it by using this scripts:

wget https://dl.fbaipublicfiles.com/grid-feats-vqa/X-152/X-152.pth

I borrowed it from here: https://github.com/facebookresearch/grid-feats-vqa#pre-trained-models-and-features

@alice-cool
Copy link
Author

alice-cool commented Sep 29, 2022 via email

@rentainhe
Copy link
Owner

When I INPUT "CUDA_VISIBLE_DEVICES=1 python extract_trar_grid_feature.py --config-file configs/X-152-grid.yaml --dataset coco_2015_test --weight_path ./weight/X-152.pth --output_dir ./data/" to extract from coco test2015.zip jpg. It throws the exception like the following:

**raise KeyError( KeyError: "**Dataset 'coco_2015_test' is not registered!

image When adding the code to fix it and entering a new window again, the error eliminates. image

But it didn't give any information about the number of instances over 80 categories.

I have no idea about this problem, cuz I think this may be specified settings in detectron2, I've try to extract GQA features using this code, however, it did not print the categories information too.

@alice-cool
Copy link
Author

When I INPUT "CUDA_VISIBLE_DEVICES=1 python extract_trar_grid_feature.py --config-file configs/X-152-grid.yaml --dataset coco_2015_test --weight_path ./weight/X-152.pth --output_dir ./data/" to extract from coco test2015.zip jpg. It throws the exception like the following:
**raise KeyError( KeyError: "**Dataset 'coco_2015_test' is not registered!
image When adding the code to fix it and entering a new window again, the error eliminates. image
But it didn't give any information about the number of instances over 80 categories.

I have no idea about this problem, cuz I think this may be specified settings in detectron2, I've try to extract GQA features using this code, however, it did not print the categories information too.

Thanks for your timely reply.

@rentainhe
Copy link
Owner

image When using multiple GPUs to run the code, the code throws the exception like the above picture...

Now I did the following modifications: ①in train_egine.py image

②in test_egine.py Just narrow tab it. image

And the error eliminate. Because of the large scale input feature, the code is running for a long time....

Thanks a lot! I've no idea if this is a bug in my code (cuz this code has not been updated for a long time), if this is a bug, would you like to submit a pull request to fix it ? : )

@rentainhe rentainhe added enhancement New feature or request help wanted Extra attention is needed labels Sep 29, 2022
@erjpc
Copy link

erjpc commented Oct 31, 2024

Hello, have you solved this problem? I want to extract 1616, but it cannot be extracted correctly. I can't reproduce the 88 I extracted myself. Thank you for your help

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

3 participants