![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
Zella Zhang
无名之辈
展开
-
Pytorch demo(三)之蚂蚁和蜜蜂
文章目录模型加载数据集并训练分类测试模型resnet1818 = 17个卷积层 + 1个全连接层流程加载数据集并训练此demo是区分蚂蚁和蜜蜂。分类测试【1】蚂蚁【2】蜜蜂原创 2020-09-19 15:54:35 · 1061 阅读 · 0 评论 -
Pytorch demo(二)之花分类(官方demo)
文章目录AlexNetdemo流程定义卷积神经网络AlexNet详解加载数据集并训练分类测试AlexNetAlexNet网络的优点:首次利用GPU进行网络加速训练使用了ReLU激活函数,而不是传统的Sigmoid激活函数以及Tanh激活函数使用了LRN局部响应归一化在全连接层的前两层中使用了Dropout随机失活神经元操作,以减少过拟合。过拟合:根本原因是特征维度过多,模型假设过于复杂,参数过多,训练数据过少,噪声过多,导致拟合的函数完美的预测训练集,但对新数据的测试集预测结果差。过度的原创 2020-09-17 14:10:29 · 661 阅读 · 0 评论 -
Pytorch demo(一)之分类测试识别
文章目录demo流程定义卷积神经网络加载数据集并训练分类测试demo流程model.py定义卷积神经网络train.py加载数据集并训练,训练集计算loss,测试集计算accuracy,保存训练模型predict.py用自己图像进行分类测试,并显示出图像改变成32*32大小后的图像和预测出的类别定义卷积神经网络def __init__(self): super(LeNet, self).__init__() self.conv1 = nn.Conv2d(3,16,5) # 第一个卷原创 2020-09-15 21:11:44 · 744 阅读 · 1 评论 -
Pytorch中Tensor与各种图像格式的相互转化(转载)
Pytorch中Tensor与各种图像格式的相互转化(原地址)转载 2020-09-09 23:10:49 · 1400 阅读 · 0 评论 -
PyTorch模型保存与加载
PyTorch模型保存与加载(原文链接)(原文更详细)常用:保存/加载 state_dict:保存torch.save(model.state_dict(), PATH)加载model = 自己的网络(*args, **kwargs)model.load_state_dict(torch.load(PATH))model.eval()转载 2020-09-07 13:00:58 · 93 阅读 · 0 评论 -
如何删除mac中的.DS_Store和git中的.DS_Store(转载)
如何删除mac中的.DS_Store和git中的.DS_Store转载 2020-09-01 16:08:08 · 170 阅读 · 0 评论 -
Pytorch使用PIL的读取单张图片并显示(转载)
原文连接转载 2020-08-31 20:50:39 · 1464 阅读 · 0 评论