图像分类
文章平均质量分 62
图像分类
Tomorrow;
这个作者很懒,什么都没留下…
展开
-
用Pytorch实现水果分类,训练+测试(内附数据集下载,完整代码)
一.数据集下载链接: https://pan.baidu.com/s/1_7blbYJc0ouCGmqe8kBnTw 提取码: c6ex 复制这段内容后打开百度网盘手机App,操作更方便哦二.训练模型1.定义数据初始化import torchvision.transforms as transformsimage_size=(224,224) #data_transforms=transforms.Compose([ transforms.RandomHorizontalFlip(),原创 2021-10-18 16:18:47 · 13811 阅读 · 17 评论 -
用Pytorch实现MNIST数据集分类(持续修改中)
1.定义数据初始化image_size=(224,224)import torchvision .transforms as transformstransform=transforms .Compose ([ transforms.ToTensor (), transforms .RandomHorizontalFlip (), transforms .Resize (image_size ), transforms.Lambda(lambda x: x.repeat原创 2021-10-25 13:39:39 · 1306 阅读 · 0 评论 -
用Pytorch实现CIFAR10数据集分类(持续修改中)
1.导包"""导包"""import collectionsimport numpy as npimport mathimport osimport shutilimport pandas as pdimport torchimport torchvisionimport matplotlib.pyplot as pltfrom torch import nnimport torch .nn.functional as F2.整理数据集"""整理数据集"""def read原创 2021-11-09 21:37:25 · 2762 阅读 · 0 评论