我有一个pytorch环境下训练好的resnet50,识别两种图片,帮我写一段代码,提取所有图片的深度学习特征,保存在一个表格里
时间: 2023-03-16 08:44:26 浏览: 156
我可以给你一段python代码,可以提取你训练好的resnet50深度学习特征,保存在一个表格里。import torch
import torchvision# 加载模型
model = torchvision.models.resnet50(pretrained=True)# 加载图片
image_list = ["image1", "image2"]# 定义一个空list,用来保存提取出的特征
feature_list = []for image in image_list:
# 将图片转换为Tensor
img_tensor = torch.tensor(image)
# 提取特征
feature = model(img_tensor)
# 将特征加入到feature_list中
feature_list.append(feature)# 将特征保存到表格中
import pandas as pd
df = pd.DataFrame(feature_list)
df.to_csv('features.csv')
阅读全文