OICR-PyTorch icon indicating copy to clipboard operation
OICR-PyTorch copied to clipboard

How is the pretrained VGG loaded?

Open oblivious137 opened this issue 3 years ago • 2 comments

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 avatar Jun 18 '22 02:06 oblivious137

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 ?

oblivious137 avatar Jun 22 '22 01:06 oblivious137

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.

ch13d2am avatar Aug 10 '22 22:08 ch13d2am