Pytorch如何用預訓練模型提取圖像特征

方法很簡單，你只需要將模型最后的全連接層改成Dropout即可。

import torch
from torchvision import models

# load data
x, y = get_data()
...


model = models.VGG16(pretrained=True)
model.classifier = torch.nn.Dropout()
feature = model(x)

over。

MARSGGBO♥原創





2019-3-5