bubbliiiing / ssd-tf2

这是一个ssd-tf2(tensorflow2)的源码,可以用于训练自己的模型。
MIT License
137 stars 36 forks source link

backbone weight names? #12

Open BuaaAlban opened 2 years ago

BuaaAlban commented 2 years ago

I have a question about how do you get the pretrained backbone vgg (You pretrained it by yourself with your model or get it from other sources)? And how can I align the weight names from other public pretrained model to the current one

bubbliiiing commented 2 years ago

imagnet?

BuaaAlban commented 2 years ago

Yes, pretrained model on imagenet. I want to replace vgg by resnet34 as backbone. Do you pretrain the vgg on imagenet by yourself?

bubbliiiing commented 2 years ago

There is pretrained model in TORCHVISION

BuaaAlban commented 2 years ago

Yes, but it's in pytorch version, though I can do the mapping of names

bubbliiiing commented 2 years ago

it's same, i get it from torchvision