resnet50的迁移学习

最新推荐文章于 2024-07-21 22:58:49 发布

易水潇潇666

最新推荐文章于 2024-07-21 22:58:49 发布

阅读量1.2k

点赞数 1

文章标签：深度学习 pytorch 图像处理

本文链接：https://blog.csdn.net/qq_40373651/article/details/118526433

版权

resnet50迁移学习

from __future__ import print_function,division
import torch
import time
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets,transforms
from torch.autograd import Variable
import numpy as np
from torchvision import models
#import matplotlib.pyplot as plt
from torch.utils.data.dataloader import DataLoader
import torchvision
from torch.utils.tensorboard import SummaryWriter

size = (224,224)
#数据类型转换
train_transforms = transforms.Compose([
   #数据增强 
   transforms.Resize(size), #将图片进行裁剪
   transforms.RandomVerticalFlip(0.5), #将图片垂直翻转
   transforms.RandomHorizontalFlip(0.3), #将图片水平翻转
   transforms.ToTensor(), #把numpy类型数据转换成tensor类型
   transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), #正则化
   transforms.RandomCrop(32,padding=4),
   transforms.CenterCrop(size=(200,200))
])
test_transforms = transforms.Compose([
    transf