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

Load pretrained model problem #33

Open
yhydhx opened this issue Jan 4, 2022 · 0 comments
Open

Load pretrained model problem #33

yhydhx opened this issue Jan 4, 2022 · 0 comments

Comments

@yhydhx
Copy link

yhydhx commented Jan 4, 2022

Thank you for your amazing work. I trained a retinanet model on my customized dataset. The detection works well on the RetinaNet model. However, I load the pretrained model in CTracker and the detection didn't work in CTracker.
I checked my code and I found that these layers are not successfully loaded in the CTracker model because of mismatched size:
#1. regressionModel.conv1.weight 256,512,3,3 => 256,256,3,3
#2. regressionModel.output.weight 36,256,3,3 => 8,256,3,3
#3. regressionModel.output.bias 8 => 36
#4. classificationModel.conv1.weight 256,512,3,3 => 256,256,3,3
#5. classificationModel.output.weight 1,256,3,3=> 9,256,3,3
I tried to change the parameters in either Ctracker model or RetinaNet model. But none of them can run successfully.
Could you tell me 1. how to train your RetinaNet with default setting? Should I change feature size from 256 to 512? Should I change the num_anchors from 9 to 1? Did you load all the weight in your RetinaNet ?

Thanks so much!
Harold.

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