13篇文章 · 3013字 · 4人关注
-
Pytorch学习笔记五——net.train与net.eval
net.train() 和 net.eval() 两个函数只要适用于Dropout与BatchNormalization的网络，会影响到训练过程...
-
Pytorch之图像分割（多目标分割，Multi Object Segmentation）示例调用预训练模型（deeplabv3_resnet101）对VOCSegmentation数据进行图像分割实验。 PyTorch的DeepLa...
-
Pytorch之目标检测（多目标检测，Multi Object Detection）示例数据 COCO 2014 http://cocodataset.org/#home[http://cocodataset.org/#home...
-
Pytorch之图像分割（单个目标分割，Single Object Segmentation）示例数据为Feta-Head-Circumference下载地址： https://zenodo.org/record/1322001#.YTH...
-
Pytorch之目标检测（单个目标检测，Single Object Detection）数据集下载地址 AMD-Training400.ziphttps://ai.baidu.com/broad/introduction[https...
-
Pytorch之图像分类（多分类，Mutli-Class Image Classification）示例数据集：STL-10数据集，pytorch的torchviso包里有提供该数据。 引入包 一些中间辅助函数 数据加载与初探 定义转换器进行数...
-
Pytorch之图像分类（二分类，Binary Image Classification）示例数据集：kaggle上面的癌症数据集Histopathologic Cancer Detection[https://www.kaggle....
-
Pytorch小技巧集
查看torch版本，系统版本 查看torch是否支持cuda（GPU），GPU数量，显示名称 device和workers全局定义 Pytorc...
-
Pytorch常见包
torch：包含所有的工具包以及tensor库 torch.nn：包含构造神经网络的一些基本模块 torch.nn.functional：包含一...
文集作者