2021
07-01
07-01
自己搭建resnet18网络并加载torchvision自带权重的操作
直接搭建网络必须与torchvision自带的网络的权重也就是pth文件的结构、尺寸和变量命名完全一致，否则无法加载权重文件。此时可比较2个字典逐一加载，详见pytorch加载预训练模型与自己模型不匹配的解决方案importtorchimporttorchvisionimportcv2ascvfromutils.utilsimportletter_boxfrommodel.backboneimportResNet18model1=ResNet18(1)model2=torchvision.models.resnet18(progress=False)fc=model2.fcmodel2.fc...
继续阅读 >