resnet提取图片特征

下载模型

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