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

How is the pretrained VGG loaded? #11

Open
oblivious137 opened this issue Jun 18, 2022 · 2 comments
Open

How is the pretrained VGG loaded? #11

oblivious137 opened this issue Jun 18, 2022 · 2 comments

Comments

@oblivious137
Copy link

The pre-trained VGG contains parameters of the two FCs after pooling. I want to know whether they are loaded before oicr training?

If so, how is this VGG pre-trained? The pretrained model seems to contain a ROIPooling layer, while the original VGG contains a GlobalPooling. Hope to point out which paper it comes from.

@oblivious137
Copy link
Author

Sorry, after checking the VGG paper, I found that the input of FC is 51277, 7=224/2^5, which is determined by the input size 224 and the number of maxpools 5.
So, are these two FCs loaded to OICR model ?

@ch13d2am
Copy link

I'm not the developer, but if you check the vgg_weights_helper module in the utils folder, you will seethat all the weights for the vgg layers are loaded. But in the vgg16.py file under models, the FCs are not included - this is exactly as it is in the OICR paper. Only the conv layers are used as feature extractor, and then passed with the selective search proposals into first SPP and then FRCNN for detection.

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