人工智能
HuaYuuuu
这个作者很懒,什么都没留下…
展开
-
如何在Pytorch中对Tensor进行shuffle
今天在训练网络的时候,考虑做一个实验需要将pytorch里面的某个Tensor沿着特征维度进行shuffle,之前考虑的是直接使用shuffle函数(random.shuffle),但是发现random.shuffle函数会导致Tensor里面数据被重复提取,进而导致输出跟输入分布发生变化。a = torch.randint(0, 2, (3, 3))print(a)random.shuffle(a)print(a)因此,经过查找资料后,总结pytorch的tensor可以利用如下.原创 2021-04-27 17:43:52 · 13366 阅读 · 1 评论 -
Denoising Prior Driven Deep Neural Network for Image Restoration阅读笔记
Denoising Prior Driven Deep Neural Network for Image Restoration为Weisheng Dong在2018年TPAMI上发表的一篇关于先验驱动的图像复原网络。这篇文章不仅仅设计了融合先验知识的图像复原网络,也在前面introduction部分和related work部分针对已有的图像复原网络进行了综述。Motivation现有的大多数基于深度学习的图像复原算法仅仅是利用深度网络直接学习一个从低像质图像到高像质图像的映射,而传统方法中的观测特原创 2020-12-14 23:07:12 · 1320 阅读 · 4 评论 -
实时语义分割脉络梳理
作为计算机视觉中的一个经典的问题,语义分割已经受到了学界的广泛关注。语义分割是针对输入的图像的每个像素,预测出该像素属于何种类别。这种相对于目标检测提供边界框信息而言,能够产生更加稠密的预测,因此提供的信息也相对于目标框更为丰富,比如能够提供目标在图像中的准确边界信息等。因此,语义分割在自动驾驶等需要精细化信息的领域上有着非常大的应用前景,能够为计算机针对场景进行理解提供重要帮助。而随着Ale...原创 2020-04-03 20:55:56 · 5458 阅读 · 1 评论 -
Pytorch Loss 无法下降的问题
最近训练自己的Landmark检测网络的时候,遇到的loss无法下降的问题,现在记录如下,一方面给自己留个记录,另一方面希望能够给大家一些参考。主要就是使用的王井东老师团队的HRNet来跑我们自己做的数据集,不得不说HRNet的Pytorch训练流程对我们来说真的特别友好,有很多细节上的优化,一旦上手之后改很多东西都很方便。之前我已经在数据集上进行了验证,然后需要添加我们自己设计的模块,...原创 2020-01-03 09:53:55 · 18473 阅读 · 7 评论