今天小编大家分享一篇pytorch实现用Resnet提取特征并保存为txt文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

接触pytorch一天，发现pytorch上手的确比TensorFlow更快。可以更方便地实现用预训练的网络提特征。 以下是提取一张jpg图像的特征的程序： # -*- coding: utf-8 -*- import os.path import torch import torch.nn as nn from torchvision import models, transforms from torch.autograd import Variable import numpy as