![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
nudt_qxx
这个作者很懒,什么都没留下…
展开
-
YOLOX自定义数据训练
有点个人笔记的意思,写的不是特别详细1.数据转成voc格式ln到dataset目录下面;2.exps/example/yolox_voc/yolox_voc_s.py下修改self.num_classes;3.data/datasets/voc_classes.py修改class name;4.yolox/exp/yolox_base.py下修改epoch值和train_size,数据较少,epoch可以改小,train_size和自己数据集大小保持一致;5.下载预训练模型yolox_m.pth6.p原创 2021-09-09 11:55:00 · 363 阅读 · 0 评论 -
Windows下安装PyTorch1.5
PyTorch简介在2017年1月18日,facebook下的Torch7团队宣布PyTorch开源后就引来了剧烈的反响。PyTorch 是 Torch 在 Python 上的衍生版本。Torch 是一个使用 Lua 语言的神经网络库, Torch 很好用, 但是 Lua 流行度不够, 所以facebook开发团队将 Lua 的 Torch 移植到了更流行的语言 Python 上,推出了PyTo...原创 2020-04-30 20:14:12 · 15434 阅读 · 9 评论 -
Pytorch加速Dataloader提升数据读取速度
出处:https://zhuanlan.zhihu.com/p/66145913在利用DL解决图像问题时,影响训练效率最大的有时候是GPU,有时候也可能是CPU和你的磁盘。很多设计不当的任务,在训练神经网络的时候,大部分时间都是在从磁盘中读取数据,而不是做 Backpropagation 。这种症状的体现是使用 Nividia-smi 查看 GPU 使用率时,Memory-Usage 占用率很...转载 2019-06-08 16:53:21 · 24310 阅读 · 4 评论 -
浅谈 PyTorch 中的 tensor 及使用
原文出处:浅谈 PyTorch 中的 tensor 及使用:https://zhuanlan.zhihu.com/p/67184419这篇文章主要是围绕 PyTorch 中的 tensor 展开的,讨论了张量的求导机制,在不同设备之间的转换,神经网络中权重的更新等内容。面向的读者是使用过 PyTorch 一段时间的用户。本文中的代码例子基于 Python 3 和 PyTorch 1.1,如果文...转载 2019-06-08 16:47:03 · 1223 阅读 · 0 评论 -
mmdetection加入SENet-154 backbone
代码搬用自:https://github.com/open-mmlab/mmdetection/issues/5881. 在mmdet/models/backbones中新建senet.py,加入以下内容:from __future__ import print_function, division, absolute_importfrom collections import Ordere...原创 2019-06-02 14:46:16 · 6251 阅读 · 10 评论 -
mmdetection自定义数据集进行训练
mmdetection对coco数据优先支持。所以在开始之前建议把自己数据修改为标准的coco格式,各种类型数据转coco格式脚本见:转换工具箱。第一步当然是定义数据种类,需要修改的地方在mmdet\datasets。在这个目录下新建一个文件,例如:my_data.py,然后把coco.py的内容复制过来,修改class类名为MyDataset最后把CLASSES的那个tuple改为自己数...原创 2019-05-04 11:57:29 · 9243 阅读 · 26 评论 -
maskrcnn benchmark自定义数据集的方法
maskrcnn benchmark自定义数据集的方法1、拷贝数据集到根目录的datasets下(和demo同级目录)如maskrcnn-benchmark/datasets/jinnan/jinnan2_round1_train_201903052、修改paths_catalog.py路径为maskrcnn-benchmark/maskrcnn_benchmark/config/pa...转载 2019-03-25 14:18:56 · 2034 阅读 · 1 评论 -
Windows下安装PyTorch1.0
PyTorch简介在2017年1月18日,facebook下的Torch7团队宣布PyTorch开源后就引来了剧烈的反响。PyTorch 是 Torch 在 Python 上的衍生版本。Torch 是一个使用 Lua 语言的神经网络库, Torch 很好用, 但是 Lua 流行度不够, 所以facebook开发团队将 Lua 的 Torch 移植到了更流行的语言 Python 上,推出了PyTo...原创 2019-01-11 14:45:12 · 31243 阅读 · 24 评论 -
Pytorch实现手写数字mnist识别
import torchimport torchvision as tvimport torchvision.transforms as transformsimport torch.nn as nnimport torch.optim as optimimport argparse# 定义是否使用GPUdevice = torch.device("cuda" if torch.cu...原创 2018-06-17 17:08:56 · 2405 阅读 · 1 评论 -
Windows下安装PyTorch0.4.0
本文系转载,出处:https://blog.csdn.net/sunqiande88/article/details/80085569。PyTorch简介在2017年1月18日,facebook下的Torch7团队宣布PyTorch开源后就引来了剧烈的反响。PyTorch 是 Torch 在 Python 上的衍生版本。Torch 是一个使用 Lua 语言的神经网络库, Torch 很...原创 2018-04-27 09:21:52 · 51942 阅读 · 27 评论 -
Pytorch从入门到精通(一):线性模型
我们先来看一个问题,然后看人工智能如何计算出最后的答案。问题很简单:一个人学习时长(单位:小时)和他成绩的对应关系如下,求出他在学习四小时后的成绩。其实这个问题一个5岁小孩都能一眼看出来,但是如何让人工智能计算出来呢。我们借助Python的numpy包,然后用梯度下降法计算出结果:import numpy as npx_data = [1.0, 2.0, 3.0]y...原创 2017-11-22 22:11:01 · 987 阅读 · 0 评论