
pytorch
文章平均质量分 55
小小飞在路上
整理一系列python机器学习、神经网络代码(主要是基于TensorFlow,pytorch)等,有时候也会分享一些自己遇到的问题及解决方法。共同学习,共同进步,更新看情况(没办法,博主实在是太懒了)!!!
展开
-
一步步完整实现VGGNet分类自己数据集(pytorch ,自己图像数据集准备,前一篇博客的细化)
1 数据准备1.1 图片路径和标签生成txt文件1.1.1 原始数据形式如下图说明:Srinivasan2014文件夹→文件夹下有Train和Test两个子文件夹→每个文件夹下有类别名称为文件夹名的子文件夹→图像数据1.1.2 生成的路径和形式1.2 代码 preprocess.pyimport osimport globimport sys sys.path.append("..") import randomdef file_txt(): BASE ='Sriniv原创 2021-04-11 20:55:54 · 2022 阅读 · 3 评论 -
pytorch加载VGG16及进行fine-tuning训练
加载VGG16模型并打印查看from torchvision import modelsnet=models.vgg16()print(net)1.1结果说明1.2查看某一部分**加载模型进行预训练,改变classifier层,固定feature层参数**2.1模型搭建import torchimport torch.nn as nnfrom torchvision import modelsfrom torchsummary import summarynet=m.原创 2020-11-18 19:34:51 · 6756 阅读 · 14 评论 -
网络结构总结1-Lenet:Lenet-5网络结构说明及pytorch实现
1.网络结构1.1论文结构1.2简化说明说明:共7层(3Conv+2pool+2FC)2.作用2.1各个部分的作用卷积:特征提取。(说明:同一卷积层中不同通道学习的是同一尺度下的多种特征,浅层卷积和深层卷积学习到的是多种尺度的特征,且深层次的卷积学习到的特征更抽象。)激活函数:引入非线性,提高模型的表达能力池化(降采样):特征压缩:降低模型复杂度防止过拟合+提取主要特征。不变性:translation(平移:不同位置),rotation(旋转:不同角度),scale(尺度:放缩)。原创 2020-11-04 17:01:02 · 1184 阅读 · 2 评论 -
VGG16网络结构图及pytorch 代码实现
1.网络结构图及对应输出结果2.pytorch代码实现import torch.nn as nnfrom torchsummary import summaryimport torchclass VGG16(nn.Module): def __init__(self): super(VGG16, self).__init__() self.maxpool1 = nn.Sequential( nn.Conv2d(3, 64, ker原创 2020-10-20 10:54:44 · 8069 阅读 · 2 评论 -
pytorch不同方法搭建CNN测试(mnist测试+自己数据集测试)
CNN搭建方法原文链接:https://www.jb51.net/article/138245.htm1.mnist数据集测试mnist数据集只是为了测试方法,效果不好,可以自己调参数,改变结构之类的。(需要注意的是自己下载的mnist数据集应该怎么建立文件进行存储然后读取)# -*- coding: utf-8 -*-"""Created on Wed May 27 23:36:25 2020@author: 小小飞在路上"""import torchimport torch.nn.转载 2020-06-09 13:49:20 · 898 阅读 · 0 评论