pytorch中torch.stack用法虽简单，但不好理解-参数列表
tensors ：为一系列输入张量，类型为turple和List dim ：新增维度的（下标）位置，当dim = -1时默认最后一个维度；范围必须介于 0 到输入张量的维数之间，默认是dim=0,在第0维进行连接 返回值：输出新增维度后的张量 情况一：输入数据为1维数据 dim = 0 : 在第0...
使用pytorch构建一个无监督的深度卷积GAN网络模型
本文为此系列的第二篇DCGAN，上一篇为初级的GAN。普通GAN有训练不稳定、容易陷入局部最优等问题，DCGAN相对于普通GAN的优点是能够生成更加逼真、清晰的图像。 因为DCGAN是在GAN的基础上的改造，所以本篇只针对GAN的改造点进行讲解，其他还有不太了解的原理可以返回上一篇进行观看。 ...
3D分割项目 | 基于Pytorch+3DUnet实现的3D体积语义分割算法
项目应用场景 用于 3D 体积语义分割场景，适用于各种物体的 3D 语义分割，比如大米、大豆的体积分割等项目效果： 项目流程 ==> 具体参见项目内README.md (1) 安装 conda install -c conda-forge mambamamba creat...
深度学习论文: Attention is All You Need及其PyTorch实现
深度学习论文: Attention is All You Need及其PyTorch实现 Attention is All You Need PDF:https://arxiv.org/abs/1706.03762.pdf PyTorch: https://github.com/shanglianl...
pytorch之model.eval、model.fuse及model.fuse.eval介绍
在 PyTorch 中，model.eval() 是用于将模型设置为评估模式的方法，而 model.fuse() 是用于量化模型中的融合操作的方法。下面是它们的详细介绍： 1. model.eval()方法介绍 当涉及到 PyTorch 中的模型评估时，model.e...
windows10 64位+cpu（无GPU） +Anaconda 安装 pytorch
现在看作者这篇文章的，已经是绕弯路了，最近pytorch推出了支持windows版本的pytorch，可以移步作者的新blog 新链接 如果对tensorflow等框架有简单了解的话，可以pip安装，输入下面两句话。不了解的话可以移步上面的链接 pip3 install htt...
深度学习 环境配置（Tensorflow,Pytorch）多版本CUDA共存
建议在anaconda下创建环境，再进行安装，方便各个环境的管理，以下安装方法在cuda和cudnn都安装好的前提下进行的。Tensorflow安装首先创建环境：conda create -n tf_0.11_gpu python=2.7； tf_0.11_gpu为环境的名称，2.7指定了pytho...
解决Pytorch数据集下载时如MNIST数据下载缓慢
解决Pytorch数据集如MNIST数据下载缓慢数据是深度学习的魂，而在download数据集时，下载了一半就停滞不前……这时看你设置的下载目录，在MNIST里会有两个文件夹：processed和raw打开raw能看到已经下载了的数据集。其实MNIST数据集很小，直接访网站，手动下载数据集：http...
超简单！用pytorch训练自己的数据集
在pytorch官网上的tutoral中，教程给了一个例子，训练一个分类器，下载CIFAR10数据集，用一个简单的CNNC网络训练模型。在我们下载的CIFAR10数据集中，已标签的数据被打包并封装在data文件夹里。我们可以在这个例子的基础上，把自己的数据放到其模型下，并实现一个简易的分类器。首先是...
60分钟入门PyTorch，官方教程手把手教你训练第一个深度学习模型
点击我爱计算机视觉标星，更快获取CVML新技术本文转载自机器之心。近期的一份调查报告显示：PyTorch 已经力压 TensorFlow 成为各大顶会的主流深度学习框架。想发论文，不学 PyTorch 怎么行？那么，入门 PyTorch 深度学习需要多久？PyTorch 的一份官方教程表示：只需要 ...
nvidia驱动、cuda和GPU版本pytorch安装
1、nvidia官网：https://www.nvidia.cn/2、找到 驱动程序 -> 所有 NVIDIA驱动程序 找到对应显卡的最新驱动3、根据自己电脑的显卡型号 查找驱动，如 GTX1050 4、找到后，下载安装即可问题：NVIDIA-SMI has failed because yo...
深度之眼Pytorch打卡（十四）：Pytorch卷积神经网络部件——卷积层与反卷积层（转置卷积层）
前言 人每天处理的信号中，有超过70%的是视觉信号，所以视觉问题，包括分类，检测，分割、风格转换等等占了深度学习任务中的很大部分。而卷积神经网络是计算机视觉领域当之无愧的霸主。卷积神经网络是稀疏连接，并且权值共享的，参数比全连接要少非常多，所以完完全全可以用图像全像素作为输入，并且它比全连接网络更...
深度学习pytorch——数据增强（持续更新）
背景介绍 大量的数据是防止过拟合的关键，但是我们如何去获取大量的数据，是自己去拍摄、录制吗？显然这种方式有极高的成本。我们可以对同一张图片进行变换得到多张图片，比如原来只有10张图片，通过变换变成了20张图片，但是并不能无限的增长，也是有限制的，这种方法就叫做数据增强。数据增强的成本基本为0，虽然最...
京东开源FastReID：用于真实人物重新识别的Pytorch工具箱
作者信息介绍 FastReID是迄今为止最完整，最高性能的工具箱，支持单个和多个GPU服务器。 最近京东开源了FastReID，作为JD AI Research中广泛使用的对象重新识别（re-id）软件系统。高度模块化和可扩展的设计使研究人员易于实现新的研究思路。友好的可管理系统配置和工程部署功能...
WGAN-GP学习笔记（从理论到Pytorch实践）
WGAN相关学习，主要文献 improved of training of WGAN首先我们需要明白一般的GAN数学表达式如下:mathminGmaxDEx∼Pr[log(D(x))]+Ex~∼Pg[log(1−D(x~))]math\mathop{min}\limits_{G}\mathop{ma...
Paperreading之三Simple Baselines for Human Pose Estimation（人体姿态coco亚军，pytorch）
本次paper是coco2018关键点检测项目的亚军方案，方法非常的简洁明了，但是效果很惊艳，达到了state of the art，paper的标题也是写了simple baseline。整篇paper包含一个sota的姿态估计和姿态跟踪，本篇博客只涉及到paper里面的姿态估计部分。前言人体...
[ 人体姿态估计 ] —— 项目学习: (1) pytorch-pose学习
模型：人体姿态估计沙漏网络（Stacked hourglass network）项目地址：https://github.com/bearpaw/pytorch-pose |--- 几点重要备忘1、LEEDS_annotations.json文件的生成：Use https://github.com/b...
Pytorch for training1——read data/image
blog torch.utils.data.Dataset create dataset with class torch.utils.data.Dataset automaticlyimport torchfrom torch.utils.data import Datasetclass MyDa...
【动手学深度学习-pytorch】8.5 循环神经网络的从零开始实现-梯度裁剪
为了防止梯度爆炸或者消失，进行梯度剪裁 @savedef train_epoch_ch8(net, train_iter, loss, updater, device, use_random_iter): """训练网络一个迭代周期（定义见第8章）""" state, timer = ...
使用 PointNet 和 PyTorch3D 进行点云分类-PointNet 的用例
PointNet 是一种高效的架构，支持许多此类应用，包括增强现实、3D 扫描等。它的多功能性和简单性使其能够支持不仅仅是分类的任务，包括（但不限于）： 3D 物体检测 点法线的预测 物体部分的分割 语义场景分割PointNet 还具有高度可扩展性，许多已发表的增强或扩展 PointNet 架构的...