![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch图像分类
文章平均质量分 97
心之所向h
这个作者很懒,什么都没留下…
展开
-
Pytorch图像分类:03使用PyTorch搭建VGG模型
使用Pytotrch搭建VGG模型原创 2024-08-02 12:50:47 · 613 阅读 · 0 评论 -
Pytorch图像分类:02使用PyTorch搭建AlexNet模型
这里将训练集和测试集的预处理函数写成了一个。"train":transforms.Compose([transforms.RandomResizedCrop(224),#随机裁剪transforms.RandomHorizontalFlip(),#随机翻转]),])在开始之前,获取了设备,查看是使用了cpu还是gpu,如果是gpu,则会打印"cuda"。原创 2024-07-29 19:56:55 · 524 阅读 · 0 评论 -
Pytorch图像分类:01使用PyTorch搭建LeNet模型
在数据进入模型之前需要进行一些预处理,例如数据中心化(仅减均值),数据标准化(减均值,再除以标准差),随机裁剪,旋转一定角度,镜像等一系列操作, 在 PyTorch 中,这些数据增强方法被放在了文件中。可以把transforms看作是一个数据预处理工具包,它里面用于处理的每个函数可以看作是单独的一个工具,我们可以用transforms.Compose将多个工具组合起来形成自己的工具包。])原创 2024-07-27 13:54:42 · 845 阅读 · 0 评论