PyTorch
ANTennaaa
这个作者很懒,什么都没留下…
展开
-
PyTorch RuntimeError: size mismatch, m1:
nn.Linear使用的时候需要给定(in_channels, num_classes),在查看torch的FastRCNNPredictor官方实现时,想弄清楚一些细节。随便打一点测试代码,报错如下RuntimeError: size mismatch, m1: [3 x 4], m2: [3 x 2] at /pytorch/aten/src/TH/generic/THTensorMath.cpp:41其实很明显,说明维度不对。于是我们就明白,FastRCNNPredictor的to..原创 2020-10-31 09:55:29 · 3895 阅读 · 0 评论 -
PyTorch VOC数据加载
这是一个读取VOC数据集的例子,供大家参考但是我想这更适合作为一个云备份,哈哈"""因为我打算先搞一下目标检测所以先用VOC这个轻量级的来做"""import torchimport torch.utils.data as dataimport numpy as npimport cv2import xml # 标注是xml格式try: import xml.etree.cElementTree as ET # 解析xml的c语言版的模块except Import.原创 2020-07-18 11:34:34 · 1724 阅读 · 4 评论 -
Pytorch 不同通道分别激活
截取出来不同的通道,进行激活即可原创 2020-07-18 00:15:49 · 371 阅读 · 0 评论 -
PyTorch 使用Visdom训练数据可视化
使用Visdom,配合浏览器在线更新各种数据,如loss以及准确度等等,甚至图像.这里提供一个简明实用的教程.安装与启动安装: pip install visdom终端启动:python -m visdom.server打开浏览器,地址栏输入http://localhost:8097/可以看到蓝色界面.如果报错需要下载我上传的一个资源进行解决:static.zip 解决visdom 浏览器蓝屏上面链接也给出了具体的操作方法编写一个最简单的绘制正弦曲线程序import .原创 2020-07-15 13:54:08 · 1079 阅读 · 0 评论 -
PyTorch 截取出某些通道
按照PyTorch的规则,经过批处理加载后的图像数据排列规则:[batchsize;channels;width;height][batchsize; channels; width; height][batchsize;channels;width;height]有些场景下我们想要提取出众多通道中的某些通道,如在目标检测场景下,我们将一些通道映射到bbox,要取出4通道数:方法:eg:取出来前两个通道target[:, :2] ...out[:, :2]...接下来就可以进行计算损.原创 2020-07-13 20:40:38 · 3699 阅读 · 0 评论 -
opencv错误显示PyTorch处理后的图像 uint8显示tensor
大坑,我在显示的时候,一个三通道图像显示出来就和二值化一样…首先给出我有问题的处理过程: rgb_img = cv2.imread(self.rgb_ids[index]) rgb_img = torch.from_numpy(rgb_img).permute(2, 0, 1).float() ... s = rgb_img[0].permute(1, 2, 0).numpy() cv2.imshow("tmd", s) .原创 2020-07-10 00:23:04 · 703 阅读 · 0 评论 -
PyTorch 预训练权重保存位置
使用Pytorch pretrained=True时,IDE下很慢,有时候还会报错。于是点击链接用浏览器下载。这时候才能看到我真实的带宽。下载完后放在哪里呢?顺藤摸瓜: if pretrained: state_dict = load_state_dict_from_url(model_urls[arch], progress=progress)通过load_state_dict_.原创 2020-05-26 11:33:13 · 5085 阅读 · 3 评论 -
PyTorch 在CPU和GPU上加载模型【转】
强大的PyTorch,允许我们在GPU训练模型,在CPU上加载。或者反过来也可以https://blog.csdn.net/dcrmg/article/details/79503978转载 2020-04-28 13:49:32 · 223 阅读 · 0 评论 -
Pytorch VGG16源码解读
我感觉我已经和时代脱轨了,有的网络已经发布好多年,已经可以视为“经典”的存在,而本菜鸟还是一知半解,刚开始了解!退而结网吧,现在开始看一下源码,了解一下经典。。。一个完成的VGG16包含如下# 摘抄自Pytorch源代码,应该不会侵权吧...# VGG类,输入特征层和类别个数,得到类别向量class VGG(nn.Module): def __init__(sel...原创 2020-04-11 16:57:53 · 2580 阅读 · 1 评论 -
Pytorch 用cfg构建网络结构
动机:使用cfg构建网络是很方便的,有必要了解一下本篇是通过参考Pytorch源码,来进行相关学习的。一直以来觉得这种方法比较抽象,但其实看一看,还是很值得的,毕竟我们生活在21世纪,接下来做研究可以借助强大的GPU算力(啥时候有钱了)构造更多深层网络,只用一个列表写出来结构显然会比一层一层手动写出来灵活性更高,修改也比较省时省力。通过阅读本文,可以学到用for循环迭代cfg列表的形式...原创 2020-04-11 15:51:43 · 2815 阅读 · 0 评论 -
Pytorch torchvision 官方数据集重定向
我想把数据集都放在同一个文件夹,包括通过torchvision下载的,我和自己搜集的其他数据集,当然,我们可以在首次选择download=True的时候指定文件夹,这样最方便。但是如果下载完了,想改文件夹,要怎么弄呢?给出过程,实测可用...原创 2020-04-10 08:45:03 · 422 阅读 · 0 评论 -
【使用报错提示】NotImplementedError进行子/父类编程
写作动机:在学习PyTorch时,有个 “_check_input_dim” 函数,该函数在子类中有,在父类 “_BatchNorm” 的forward中只有类似声明一样的self._check_input_dim(input)。 这时候在父类中无法转到定义,于是有此文。以下是正文NotImplementedError用在子父类编程中。首先给出我们的实例代码,然后进行分析。...原创 2020-03-02 23:56:19 · 525 阅读 · 0 评论