resnet提取图片特征

原创

已于 2023-05-31 09:54:46 修改 · 1k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python

于 2023-05-30 14:21:56 首次发布

下载模型

wget https://digix-algo-challenge.obs.cn-east-2.myhuaweicloud.com/2020/cv/6rKDTsB6sX8A1O2DA2IAq7TgHPdSPxJF/train_data.zip -o train_data.zip

ResNet提取图片特征向量



from torch.autograd import Variable
import torch
import torch.nn as nn
import torchvision.transforms as transforms
import pretrainedmodels
from PIL import Image

import ssl
ssl._create_default_https_context = ssl._create_unverified_context

TARGET_IMG_SIZE = 224
img_to_tensor = transforms.ToTensor