用resnet50提取图片的特征向量

最新推荐文章于 2025-04-13 22:59:45 发布

Rashore

最新推荐文章于 2025-04-13 22:59:45 发布

阅读量8.6k

点赞数 10

分类专栏： python 机器学习文章标签：机器学习人工智能 pycharm

本文链接：https://blog.youkuaiyun.com/ruoruoruo666/article/details/121385680

版权

依次读取文件夹中所有的图片，最后将图片保存为npy文件。

import os.path
import torch
import torch.nn as nn
from torchvision import models, transforms
from torch.autograd import Variable
import numpy as np
from PIL import Image

features_dir = './features'  # 存放特征的文件夹路径
data_list = []# 定义一个空的存放数据的列表
path='./RSICD_images'#这里是存放图片的路径，所有的图片均存放RSICD_images中
for filename <