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

AssertionError #3

Open
tutu-star opened this issue Oct 31, 2022 · 5 comments
Open

AssertionError #3

tutu-star opened this issue Oct 31, 2022 · 5 comments

Comments

@tutu-star
Copy link

tutu-star commented Oct 31, 2022

How can I get 'pvt_v2_b2_li.pth' ? I can't find this file.
image

@GuangxingHan
Copy link
Owner

The pre-trained PVTv2 models are here https://github.com/whai362/PVT/tree/v2/classification#model-zoo

@tutu-star
Copy link
Author

Hello, I have a question. After downloading your pre training model, I want to know how to use your code for testing.

@GuangxingHan
Copy link
Owner

The pre-trained model is after the second step training (two-branch based model over base classes). You can directly use the script of few-shot fine-tuning for evaluation. For example, on COCO dataset, you can use the following script

sh scripts/two_branch_few_shot_finetuning_coco_pvt_v2_b2_li.sh

@tutu-star
Copy link
Author

Hello, when I use my own dataset for evaluation, the following problems are reported. How can I solve them?
image

@GuangxingHan
Copy link
Owner

It seems that the direct cause is that dataset_name is not an attribute of meta.
You can follow the coco_evaluation or pascal_voc_evaluation to implement the evaluation script for your dataset. Actually according to Line 42 in pascal_voc_evaluation, we use meta.dirname rather than meta.dataset_name.

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