![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 61
跑酷托托
这个作者很懒,什么都没留下…
展开
-
pytorch之数据集构造
这些天看的东西,真的是比较多,相比以前来说,对我的学习方式起到颠覆性作用。我目前觉得,我们学到的东西,更多是孤立的,因此,在吸收一定知识后,需要在脑子里形成知识体系。需要把自己以前学到的东西进行整理,形成一个体系,这篇文章讲解的是,深度学习中pytorch数据集的构造!!!pytorch中有两个自定义管理数据集的类,torch.utils.data.DataSettorvchvision....原创 2019-07-07 19:46:08 · 1892 阅读 · 0 评论 -
pytorch教程一Ignite
python是一种面向对象抽象,书写深度学习程序,最好是将所有程序对象像化,这样的化,可以花更多的精力在网络结构上,在这里介绍基于pytorch高级库Ignite,这个非常有用的库,下面看下源码布局Ignite介绍这个package中包含四个包:contribenginehandlersmetrics这里每个package的源码有兴趣,可以读一读,感受一下,大神是如何设计pack...原创 2019-07-02 19:18:29 · 3362 阅读 · 0 评论 -
pytorch三——data
总览data文件一、datasetdataset:负责定义个数据库文件,__init__.py 空文件,申明dataset文件是packageimport osimport numpy as npfrom PIL import Imagefrom torch.utils import data# 根据root,读每个文件的地址def read_images(root, t...原创 2019-07-04 10:05:58 · 369 阅读 · 0 评论 -
pytorch三——slover
总览sloverslover这个文件负责 定义optimizer, 对optimizer超参进行设计。一、build.pyimport torchdef make_optimizer(cfg, model): params = [] for key, value in model.named_parameters(): # model.named_paramet...原创 2019-07-04 10:16:53 · 1075 阅读 · 2 评论 -
pytorch四——layers
总览layerslayers负责自定义网络层和损失函数。一、conv_layers.py定义卷积层+初始化from torch import nndef conv_layer(in_channels, out_channles, kernel_size, stride=1, padding=0, bias=True): layer = nn.Conv2d(in_chan...原创 2019-07-04 10:24:01 · 1184 阅读 · 0 评论 -
pytorch五——modeling
总览modeling由于论文中有fcn8s、16s、32s对比实验,且前部分均和vgg16相同,故定义了backbone文件一、backbonevgg.py这里也不多说了,预训练模型,接下来需要接受~import torchimport torchvisionfrom torch import nnclass VGG16(nn.Module): def __ini...原创 2019-07-04 11:01:17 · 504 阅读 · 1 评论 -
pytroch五——engine
总览engineinference.py : 测试时运行trainer.py: 训练模型时运行trainer.py完整代码见链接:这里为了简单,赋值命令和logging操作,SummaryWriter操作就会省去:一、importimport loggingimport matplotlib.pyplot as pltimport numpy as npimport ...原创 2019-07-04 11:47:38 · 3246 阅读 · 6 评论