有时,我们需要利用预训练网络权重提取全连接前的特征。需要将全连接去除掉,然后后面根据自己的任务进行修改。
self.feature_extraction = models.resnet34(pretrained=True)
self.feature_extraction.fc = nn.Sequential() # remove fc
或者,在设计网络层面,将对应的fc去除掉,然后按对应键对进行加载即可。
有时,我们需要利用预训练网络权重提取全连接前的特征。需要将全连接去除掉,然后后面根据自己的任务进行修改。
self.feature_extraction = models.resnet34(pretrained=True)
self.feature_extraction.fc = nn.Sequential() # remove fc
或者,在设计网络层面,将对应的fc去除掉,然后按对应键对进行加载即可。