VGG16 Imagenet weight initialization

lichengunc / mask-faster-rcnn

Mask R-CNN

59 stars 30 forks source link

VGG16 Imagenet weight initialization #5

Closed mees closed 4 years ago

mees commented 5 years ago

Hi, which VGG16 Imagenet pretrained pytorch model can we use to intialize your VGG16 network? I tried converting the original VGG16 caffemodel to pytorch, but then there were some keys missing in the state dict. Thanks in advance!

mees commented 5 years ago

Also I read that some some pretrained VGG16 models expect the input in RGB, others in BGR etc. how is it for your implementation? Which pretrained VGG16 model should we use for your code?