2019年3月5日
Pytorch如何用预训练模型提取图像特征
摘要： 方法很简单，你只需要将模型最后的全连接层改成Sequential即可。 import torch from torchvision import models # load data x, y = get_data() ... model = models.VGG16(pretrained=True 阅读全文
posted @ 2019-03-05 21:29 marsggbo 阅读(3546) 评论(0) 推荐(0) 编辑