2020年2月14日

pytorch(6)----数据处理

摘要： 主要内容 1、 主流的图像数据集 imageNet 、 PASCAL VOC 、COCO、KITTI、Cityscape、Udacity 2、pytorch 数据加载过程 继承Dataset类 > 增加数据变换 > 继承Dataloade 3、继承torch.utils.data.Dataset 抽 阅读全文

posted @ 2020-02-14 20:51 feihu_h 阅读(320) 评论(0) 推荐(0)

pytorch(5)----模型加载保存、网络模型库 torchvision.models

摘要： torchvision.models 提供了如：VGG、ResNet、Inception 等众多经典的网络模型结构。 安装： pip install torchvision 或 conda install torchvision import torch from torch import nn f 阅读全文

posted @ 2020-02-14 18:11 feihu_h 阅读(590) 评论(0) 推荐(0)

pytorch(4)----nn.Module、nn.functional、nn.Sequential、nn.optim

摘要： 基本内容 1、 实现一个网络模型，需要继承torch.nn.Module类，初始化模型参数，并实现forward()函数 2、torch.nn.functional 提供一些如损失函数等不需要学习参数的功能函数 3、torch.nn.Sequential() 模块简化简单模型的定义 4、损失函数to 阅读全文

posted @ 2020-02-14 13:51 feihu_h 阅读(553) 评论(0) 推荐(0)

导航