![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch笔记
文章平均质量分 66
MC.zeeyoung
这个作者很懒,什么都没留下…
展开
-
在加载theano框架时遇到的gcc加载问题
当在import theano 的时候,会报这个的错误:cc1.exe: sorry, unimplemented: 64-bit mode not compiled in不要怀疑,一定时你mingW没有安装好,我再找了一下午之后,终于再这个网页找到了答案思路https://stackoverflow.com/questions/33850090/compilation-issue-when-running-theano我的安装姿势是这样的:下载链接:https://udomain.dl.sour原创 2021-10-20 17:08:42 · 458 阅读 · 0 评论 -
Export to ONNX Record
First of all, we need weights belongs to our export model, for example resnet.pththen we load this weight and model structure by this code原创 2021-07-07 11:18:54 · 198 阅读 · 0 评论 -
YOLOV5代码解析(小白系列一)
yolov5 代码脚本解析1. train.py因为本人太菜了,所以此处就通过一行行debug,然后先记录整个运行逻辑,然后后面再一点点增补每个trick的原理。def main(opt): setlogging(RANK) if RANK in [-1,0]: print(colorstr('train: ')+', '.join(f'{k}={v}' for k,v in vars(opt).items())) check_git_status() check_requiremen原创 2021-07-02 17:31:18 · 7444 阅读 · 6 评论 -
Cascade rcnn在linux环境下运行的调试记录(pytorch版本)
这个cascade RCNN的调试是真的折磨人,对于新手的我,查了各种各样奇奇怪怪的资料,终于调通了。以此记录,如果能帮助到大家,那就更好了。调试步骤第一步 下载工程第二步 观察setup.py第三步 配置环境报错记录第一步 下载工程我的工程目录是这样的此处我们默认大家可以正确导入数据集,因为此处主要记录的是如何生成nms里面调用的_C库。第二步 观察setup.py用命令行进入 到 lib文件夹那么我们此时,先别急着运行什么,我们先打开setup.py,看看里面的头文件结构是什么从这原创 2021-04-08 10:10:41 · 1105 阅读 · 7 评论 -
Pytorch学习Day05[连载]
记录权值初始化技巧一、发现问题二、解决问题1、无激活函数2、tanh激活函数3、relu激活函数一、发现问题在搭建网络后,有时候会出现梯度爆炸或梯度消失,这是因为每个输出节点的标准差过大或者过小先上源码import torchimport torch.nn as nnimport randomimport numpy as npdef set_seed(seed=1): random.seed(seed) np.random.seed(seed) torch.man原创 2020-05-18 15:22:52 · 117 阅读 · 0 评论 -
Pytorch学习Day04[连载]
Transform专场——图像增强一、transforms——变换1、Crop(裁剪)1)CenterCrop2)RandomCrop3)RandomResizedCrop4)FiveCrop5)TenCrop2、Flip(翻转)1)Horizontal Flip2)Vertical Flip3)RandomRotation3、Pad(填充)4、ColorJitter(图像颜色调整)5、Grays...原创 2020-05-07 18:18:35 · 213 阅读 · 0 评论 -
Pytorch学习Day03[连载]
纸币二分类——引入dataloader模块使用一、数据准备1、划分数据集step1 导包step2 判断新建数据集文件是否存在step3 划分数据集以及建立文件夹路径step4 划分三个集合的配比step5 查看路径下分类文件情况step6 计算各个数据集容量step7 根据既定数据量划分数据集路径step8 组装数据集路径2、在训练脚本关联数据集以及数据预处理step1 设置超参step2 设置...原创 2020-05-05 10:26:58 · 249 阅读 · 1 评论 -
Pytorch学习Day02[连载]
防止自己真的忘掉,记了再说一、张量的操作1、cat 张量拼接2、stack 创建新维度,扩展维度3、chunk 将张量按维度dim进行平均切分4、split 将张量按维度dim进行切分5、index_select() 按index索引数据6、masked_select() 筛选数据,按照mask中的True进行索引7、reshape 变换张量形状8、torch.transpose() 交换维度9、...原创 2020-05-02 23:22:55 · 206 阅读 · 0 评论 -
Pytorch学习Day01[连载]
Pytorch学习笔记一、Tensor关系二、Tensor创建方式1、直接法2、依据数值创建1)通过torch.from_numpy 创建张量2)通过torch.zeros创建张量3)通过torch.full创建全1的张量4)通过torch.arange() 创建等差的1维张量5)通过torch.linspace() 创建均分数列张量6)通过torch.eye() 对角矩阵3、依据概率创建1)通过...原创 2020-04-28 15:40:48 · 133 阅读 · 0 评论