Hello!
I want to get 2048 features from a picture by using pretrained resnet-50,is it ok by these code?
resnet50_feature_extractor = models.resnet50(pretrained = True)
resnet50_feature_extractor.fc = nn.Linear(2048, 2048)
nn.init.eye_(resnet50_feature_extractor.fc.weight)
Thanks for your help!