深度学习笔记
weixin_43710224
这个作者很懒,什么都没留下…
展开
-
深度学习笔记——pytorch学习02——自定义数据集与Sequential模块
1Dataset定义 1.1Dataset 自定义Dataset继承torch.utils.data.Dataset类,创建这个类需要实现三个方法: ** __init __**: 主要用于创建对象,在此方法中一般需要传入图片的路径,通过os模块读取所有图片的路径,同时完成对训练、实验、测试集的划分; #读取所有图片的路径 imgs=[os.path.join(root,img) for img in os.listdir(root)] __getitem __: 主要实现对data、label的获取,原创 2020-07-19 10:29:43 · 343 阅读 · 0 评论 -
深度学习笔记——pytorch学习01——简单神经网络的练习
1.Le-Net5神经网络流程回顾 Le-Net5神经网络模型如下:(图片来自Yann LeCun的论文) 本练习流程如下: 32323→conv1(3655)→28286→relu→maxpool(22)→14146→conv2(61655)→101016→relu→maxpool(22)→5516→view(-1,5516)→1400→fc1(400,120)→1120→relu→fc2(120,84)→1*84→relu→fc3(84,10)→softmax(10) 2.神经网络的搭建 2.1创建神原创 2020-07-01 09:35:25 · 528 阅读 · 0 评论