Pytorch
夏天的欢
技术不重要、关键是要骚
展开
-
pytorch使用FocalLoss损失函数用于分类问题
参考:https://zhuanlan.zhihu.com/p/28527749https://www.jianshu.com/p/30043bcc90b61、创建FocalLoss.py文件,添加一下代码代码修改处:classnum 处改为你分类的数量P = F.softmax(inputs) 改为 P = F.softmax(inputs,dim=1)import torchimport torch.nn as nnimport torch.nn.functional as Ffr原创 2021-01-20 16:30:03 · 6326 阅读 · 6 评论 -
pytorch修改网络结构后如何加装预训练模型,以resnet为例
如果不知道如何使用预训练模型训练的朋友,使用resnet18模型训练自己的数据集的教程请看这篇博客一、如果我们网络没任何修改还是训练imagenet的数据集,1000个类别,那么只需以下代码:model_ft = models.resnet18(pretrained=True)model_ft = model_ft.to(device)二、如果只修改训练自己数据集类别如果我们训练自己的数据集,假如自己的数据集只有10个类别,那么需先加载预训练模型,然后修改最后的全连接层,改成以下代码加载预训练原创 2021-01-15 11:17:43 · 8476 阅读 · 6 评论 -
window10下pytorch解决不能使用多进程问题
问题:window10 pytorch不能使用多进程,num_workers只能设置为0,导致GPU训练时速度太慢。查阅网上资料,都window10不能使用多进程,其实是可以的。解决:在你的代码里加上if __name__ == "__main__":例如:if __name__ == "__main__": train_model(model_ft, criterion, optimizer_ft, exp_lr_scheduler,num_epochs)这样就可以使用多进程了,原创 2021-01-02 13:53:36 · 5269 阅读 · 6 评论 -
国内源安装pytorch
pytorch官网:https://pytorch.org/选择对应版本建议使用 pip 安装加上国内源,注意国内源一定要是https域名的,很多博客的都是http的,都没用,弄了我一晚上,坑爹。清华:https://pypi.tuna.tsinghua.edu.cn/simple阿里云:https://mirrors.aliyun.com/pypi/simple/中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/华中理工大学:https://py原创 2020-05-27 00:51:57 · 9563 阅读 · 1 评论 -
PyTorch 进阶学习(二)————STN:空间变换网络(Spatial Transformer Network)
文章目录数据集的加载空间变换网络的介绍定义网络训练和测试模型可视化 STN 结果官方文档地址:https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html在本教程中,您将学会如何使用 空间变换网络 的视觉注意力机制来扩充网络。如果需要了解更多 空间变换网络 可以在 DeepMind 论文中详细阅读。空...翻译 2019-12-14 22:10:50 · 5677 阅读 · 4 评论 -
PyTorch 进阶学习(一)————迁移学习进行图像分类(ResNet)
文章目录数据集的加载定义训练函数可视化模型预测使用微调 ConvNet 的方法训练和评估ConvNet 作为固定特征提取器训练和评估官方文档:https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html本教程中,您将学会如何使用 迁移学习 来训练卷积神经网络进行图像的分类。注释:实际上,很少有人从头开始训练整个...翻译 2019-12-13 16:58:44 · 6174 阅读 · 12 评论 -
PyTorch 入门学习(八)————torchvision.models
待写原创 2019-12-11 22:09:16 · 698 阅读 · 0 评论 -
PyTorch 入门学习(七)————torchvision.transforms
文章目录管理各个transform,使用Compose一、裁剪 ------ Crop1、随机裁剪:transforms.RandomCrop2.中心裁剪:transforms.CenterCrop3.随机长宽比裁剪 transforms.RandomResizedCrop4.上下左右中心裁剪:transforms.FiveCrop5.上下左右中心裁剪后翻转: transforms.TenCrop...转载 2019-12-11 21:29:31 · 1993 阅读 · 0 评论 -
PyTorch 入门学习总览
文章目录数据读取1、数据的读取方式2、数据预处理对于深度学习框架,训练模型时主要的代码包含3个方面:数据读取、网络构建、其他设置。基本上掌握了这3方面就能够很好的使用这个框架。Pytorch也不例外。Pytorch 官方入门教程:https://pytorch.org/tutorials/Pytorch 的官方 GitHub 地址:https://github.com/pytorch/pyt...原创 2019-12-11 20:43:02 · 225 阅读 · 0 评论 -
PyTorch 入门学习(六)————编写自定义数据集,数据加载器和转换
编写自定义数据集,数据加载器和转换解决任何机器学习问题都需要花费大量精力来准备数据。PyTorch提供了许多工具来简化数据加载过程,并使代码更具可读性。在本教程中,我们将看到如何从自定义的数据集中加载和预处理/数据增强。确保安装了以下工具包:scikit-image:用于图像输入/输出和变换pandas:为了更轻松地解析csv一、使用ImageFolder数据集的格式时torchv...原创 2019-12-11 16:22:16 · 1954 阅读 · 2 评论 -
PyTorch 入门学习(五)————数据并行处理(多GPU)
原文地址:https://pytorch.org/tutorials/beginner/blitz/data_parallel_tutorial.html#sphx-glr-beginner-blitz-data-parallel-tutorial-py翻译 2019-12-11 10:49:31 · 320 阅读 · 0 评论 -
PyTorch 入门学习(四)————训练分类器
原文链接:https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#sphx-glr-beginner-blitz-cifar10-tutorial-py数据处理通常,我们处理的数据有图像、文本、音频或者视频数据,可以使用python的标准包将数据加载到numpy数组中,然后将此数组转换为torch.*Tensor。...翻译 2019-12-09 21:22:19 · 725 阅读 · 0 评论 -
config key: DATASETS.NAMES
报错:Traceback (most recent call last): File "tools/train.py", line 80, in <module> main() File "tools/train.py", line 34, in main cfg.merge_from_list(args.opts) File "/home/huan/an...原创 2019-11-17 16:08:43 · 3171 阅读 · 4 评论 -
行人重识别的代码复现
参考:https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial1、环境的安装系统的基础环境:ubantu16.04CUDA9.0+Cudnn7.4.2Python3.7.4Anaconda 3创建虚拟环境conda create -n re_id python=3.7.1source...原创 2019-11-15 18:36:23 · 6411 阅读 · 24 评论 -
RuntimeError: Error(s) in loading state_dict for ft_net:
一、我出现这个错误的原因是:我在加载之前的模型,想进行再次训练,但是model.load_state_dict(torch.load(“要加载的路径”))这一步就开始报上面的错误。原因是我的模型之前加了 PCB,但是现在重新训练的代码要求不用PCB,导致之前的model不适用现在的代码环境出错。解决:要么将现在的代码PCB那句话改成用PCB,要么将之前的model重新训练不加PCB的,适合...转载 2019-11-14 20:40:39 · 2547 阅读 · 0 评论 -
PyTorch 入门学习(三)————搭建神经网络
神经网络可以使用该torch.nn软件包构建神经网络现在,您已经了解了autograd,这nn取决于 autograd定义模型并对其进行区分。一个nn.Module包含层,和输入值forward(input),返回值output。例子:以下对数字图像进行分类的网络这是一个简单的前馈网络。输入–隐藏层–输出层神经网络的典型训练过程如下:定义具有一些可学习参数(或权重)的神经网络...翻译 2019-11-05 10:05:45 · 697 阅读 · 0 评论 -
PyTorch 入门学习(二)————Autograd:自动微分
https://pytorch.org/tutorials/beginner/blitz/autograd_tutorial.html#sphx-glr-beginner-blitz-autograd-tutorial-py该autograd软件包是PyTorch中所有神经网络的核心。让我们首先简要地访问它,然后我们将去训练我们的第一个神经网络。该autograd软件包可自动区分张量上的所有...翻译 2019-11-03 16:36:19 · 356 阅读 · 0 评论 -
PyTorch 入门学习(一)————什么是PyTorch?
What is PyTorch?PyTorch是一个基于python的科学计算软件包,主要用于:代替Numpy以使用GPU的功能提供最大灵活性和速度的深度学习研究平台Tensors张量类似于Numpy的ndarrays,并且可以在GPU上使用加速计算from __future__ import print_functionimport torch1、矩阵的创建构建一个未初始...翻译 2019-11-03 10:38:21 · 279 阅读 · 0 评论