PyTorch官方文档学习笔记
文章平均质量分 94
薄的算法之芯
不会搞艺术的工程师不是个好厨子(专注于分享机器学习,深度学习领域文章。)
展开
-
TORCHVISION 目标检测微调教程
TORCHVISION 目标检测微调教程前言前四节,我们初步掌握了通过PyTorch构建神经网络模型,以及优化参数,模型集成等问题,本章程将微调Penn-Fudan 数据库中用于行人检测和分割的预训练Mask R-CNN模型。它包含 170 张图像和 345 个行人实例,我们将用它来说明如何使用 torchvision 中的新功能在自定义数据集上训练实例分割模型。定义数据集用于训练对象检测、实例分割和人物关键点检测的参考脚本允许轻松支持添加新的自定义数据集。数据集应该继承自标准 torch.ut原创 2021-08-08 12:00:47 · 829 阅读 · 0 评论 -
优化模型参数,模型集成
前言前三节我们掌握了模型和数据,这节我们将通过优化数据参数来训练,验证和测试模型。训练模型是一个迭代过程;每次迭代(称为epoch),模型对输出进行预测,计算其预测误差(loss),收集误差相对于其参数的导数,使用梯度下降优化参数Prerequisite Code首先从前两节中Datasets & DataLoaders和Build Model部分加载代码import torchfrom torch import nnfrom torch.utils.data import Data原创 2021-08-06 09:31:55 · 448 阅读 · 0 评论 -
使用PyTorch构建神经网络以及反向传播计算
使用PyTorch构建神经网络以及反向传播计算前一段时间南京出现了疫情,大概原因是因为境外飞机清洁处理不恰当,导致清理人员感染。话说国外一天不消停,国内就得一直严防死守。沈阳出现了一例感染人员,我在22号乘坐飞机从沈阳乘坐飞机到杭州,恰好我是一位密切接触人员的后三排,就这样我成为了次密切接触人员,人下飞机刚到杭州就被疾控中心带走了,享受了全免费的隔离套餐,不得不说疾控中心大数据把控是真的有力度。在这一段时间,也让我沉下心来去做了点事,之前一直鸽的公众号也开始写上了。。。不过隔离期间确实让我这么宅的人都感原创 2021-08-05 00:31:26 · 869 阅读 · 0 评论 -
基于PyTorch构建手写识别
使用PyTorch构建神经网络模型进行手写识别PyTorch是一种基于Torch库的开源机器学习库,应用于计算机视觉和自然语言处理等应用,本章内容将从安装以及通过Torch构建基础的神经网络,计算梯度为主要内容进行学习。How can we install Torch?Torch在Linux,Windows,Mac等开发环境下都有特定的安装方法,首先搜索官方网页https://pytorch.org/,由下图所示我们可以根据自己适合的环境进行选择,我使用的是1.9.0版本Windows环境下con原创 2021-08-04 21:47:52 · 860 阅读 · 0 评论