- 博客(7)
- 收藏
- 关注
原创 Pytorch中inplace操作
之前在写训练代码时,遇到了inplaceoperation导致的问题,所以就了解了一下pytorch的inplaceoperation,在此记录一下inplace操作的一些知识。稍微了解过python的人应该会知道python的变量名是一个类似索引的东西,其指向内存中的一个对象。而对该变量重新赋值,实际上是将该变量名指向内存中的其它对象,原对象本身其实并未改变。而Inplace操作并非如此,该操作会直接改变原对象的内容,这样能减少内存的消耗,但也会带来一些隐患。...
2022-07-21 19:59:40
4133
原创 颜色空间与颜色距离
问题描述在做发色迁移的任务时,需要对比修改后图片的发色与指定图片的发色,此时需要在颜色空间中对两张图片进行对比,但问题出现了,如何计算两张图片的颜色差异呢?颜色空间RGB颜色空间RGB颜色空间以R(Red:红)、G(Green:绿)、B(Blue:蓝)三种基本色为基础,进行不同程度的叠加,产生丰富而广泛的颜色,所以俗称三基色模式。在大自然中有无穷多种不同的颜色,而人眼只能分辨有限种不同的颜色,RGB模式可表示一千六百多万种不同的颜色,在人眼看来它非常接近大自然的颜色,故又称为自然色彩模式。红绿蓝代表
2022-03-30 22:26:07
3873
3
原创 “Ninja is required to load C++ extensions”解决方案
问题描述Ninja is required to load C++ extensions在跑一份代码时,由于该代码中需要调用 torch/utils/cpp_extension.py 文件,而此时又未安装ninja库,所以会出现如下错误:RuntimeError: Ninja is required to load C++ extensions此时问题只在于未安装ninja库,解决方法如下:pip install ninjaPS 什么是Ninja?在传统的 C/C++ 等项目构建时,通
2022-03-29 20:22:07
46597
4
原创 with torch.no_grad():显著减少测试时显存占用
问题描述 将训练好的模型拿来做inference,发现显存被占满,无法进行后续操作,但按理说不应该出现这种情况。RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 7.93 GiB total capacity; 6.94 GiB already allocated; 10.56 MiB free; 7.28 GiB reserved in total by PyTorch)解决方案...
2021-10-17 19:32:42
4556
原创 机器学习数学基础——群论
群论的定义对于群论是什么,这里引用百度百科中的一段介绍:群论,是数学概念。在数学和抽象代数中,群论研究名为群的代数结构。群在抽象代数中具有基本的重要地位:许多代数结构,包括环、域和模等可以看作是在群的基础上添加新的运算和公理而形成的。群的概念在数学的许多分支都有出现,而且群论的研究方法也对抽象代数的其它分支有重要影响对于我本人而言,看完这段描述依然很懵,有一种听君一席话,如听一席话的感觉,压根还是不知道群论是个啥。但是没关系,看完这里我们只要知道有群论这个东西...
2021-10-10 21:00:56
4138
原创 FFmpeg学习笔记
前言 对于视频的读取与处理,经常会碰到opencv与ffmpeg,所以本文就记录一下ffmpeg的学习过程,以及解答之前的几个疑惑:opencv与ffmpeg的区别是什么,什么时候使用后opencv,什么时候使用ffmpeg?ffmpeg的安装过程是什么,该怎么使用它?目前python有哪些方法使用ffmpeg,各有什么优缺点?FFmpeg介绍 以下是百度百科关于ffmpeg的介绍:FFmpeg是一套可以用来记录、转换数字音频、视频,并能将其转化...
2021-09-25 14:05:31
3105
原创 Pytorch随笔--Tensor与Variable
前言该文章是我回过头来看之前的pytorch笔记的一些记录,以自己的思路将一些相关的内容集合起来进行记录,算是对之前阶段学习的总结。问题描述刚接触pytorch的时候,一个很常见的操作就是参考别人的代码或者观看别人的教学视频。但随着pytorch的更新,一些特性便会发生变化,这就很明显地会体现在代码上,比如之前的一些代码和现在的代码将会变得不一样,尽管它们实现的功能基本是一样的。尤其是对于训练模型这一部分来说,基本的流程会很相似,但几年前的代码和当前的代码却会有很多地方不同,这对于初学者来说会是
2021-09-22 21:28:23
1092
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人