基于Pytorch框架的花卉图像识别

本文章记录一个图像分类任务和一些训练策略。
涉及到数据如何预处理和模型的搭建,怎么保存怎么测试怎么输出,使用卷积神经网络去做
在这里插入图片描述
在这里插入图片描述

模块torchvision中的transforms函数用来进行数据增强–也可在数据量较少的情况下使用(本项目源码中大概6000个测试集),而引入的models包即使用现成的模型(例如CV中比较核心的模型reshet),而引入的datasets将数据和标签读取出来

1、数据预处理

接下来解释预处理的各个步骤以及含义:
Compose按顺序组合每一步操作,在预处理中做的第一件事是使用Resize(h,w)将图像转变为相同的尺寸大小,但可能会丢失信息,且h和w的值会显著影响最终结果,要通过某种策略选好,使用卷积时值大相对于值小来说卷积的地方也就越多,计算越慢,卷积的层数越大也就越慢
数据增强,可用于数据量不够,通过平移、旋转等操作来对有限的数据加大数据集的数量
数据增强操作有:
当中一个常见操作RandomRotation(x),做一个随机的旋转,x代表旋转的角度-x~x之间
CenterCrop(x)裁剪,将图片裁剪

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值