目录
2.2 Medium Baseline (acc>0.73207)
2.3 Strong Baseline (acc>0.81872)
2.4 Boss Baseline (acc>0.88446)
Machine Learning HW3
任务
图像分类
数据下载
百度网盘:
链接:https://pan.baidu.com/s/1gEiw4nIYDA4puMIhqBI5Og?pwd=pwyl
提取码:pwyl
结果
全过strong baselin,public score与bossline差0.2
改进方法
2.1 Simple line (acc>0.50099)
运行课程上给的基础代码
2.2 Medium Baseline (acc>0.73207)
对数据集进行数据增强,并训练更长的时间,这里的图像变换,一是可以增强模型的鲁棒性,二是可以用于进一步扩充数据集,使得我们的训练集得到扩充。
test_tfm = transforms.Compose([
transforms.Resize((128, 128)),
transforms.ToTensor(),
])
# 对于训练图片做处理,数据增强,常用的几种有效数据增强方法:
train_tfm = transforms.Compose([
# Resize the image into a fixed shape (height = width = 128)
transforms.Resize((128, 128)),
# You may add some transforms here.
transforms.RandomHorizontalFlip(p=0.5), # 50%的概率水平翻转
transforms.RandomVerticalFlip(p=0.5), # 50%的概率垂直翻转
transforms.RandomCrop(128, padding=10),#填充剪裁
# transforms.RandomGrayscale(p=0.1) #根据概率转灰度channel=1,CNN中in_channel=3,不可行
transforms.ColorJitter(brightness=0.5, con