resnet18模型 批量提取文件夹中所有图片的某一层特征 数据为numpy类型,保存为npy文件
'''批量提取文件夹中所有图片的某一层特征 数据 numpy类型'''
import torch
import torchvision.models as models
import numpy as np
import torch.nn.functional as F
from torchvision import models, transforms
from PIL import Image
from torch.autograd import Variable
import torch.nn as nn
from PIL import Image
import os
import re
import pandas as pd
import datetime
extract_list = ["conv1", "maxpool", "layer1", "layer2", "layer3", "layer4", "avgpool", "fc"]
img_path = "D:/" # 输入文件夹路径
saved_path = "D:/" # 输出文件夹路径
names = os.listdir(img_path)
resnet = models.resnet18(pretrained=False)
resnet.load_state_dict(torch.load('./resnet18-f37072fd.pth'))
# print(resnet) #模型结构
transform = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTen