自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(20)
  • 收藏
  • 关注

原创 anomalib实战学习记录:安装及训练官方数据

源码下载地址:我下载的版本是:anomalib-1.1.0操作系统是:win11代码的结构为:新版本的anomalib没有单独的train.py文件了,后面需要自己新建train.py文件,或者用命令行进行训练。

2024-07-17 11:17:09 619

原创 GhostNet学习记录:

简单的记录一下对ghostnet的学习。1、要解决的问题:作者提出,在一个训练好的神经网络中,会存在很多的冗余特征,如下图(图出自论文),而这些冗余的特征可以通过一些廉价的操作从别的特征图变换中得到。因此作者的想法是,通过卷积得到一部分的特征,然后通过一些廉价的操作从这些特征图中得到冗余的特征。2、ghost卷积:卷积过程:(1)先通过卷积生成一部分的特征图。(2)对每个通道通过廉价的线性变换(其实就是对每个通道进行一次卷积,等价于深度卷积)。

2024-07-09 14:58:41 402

原创 关于 model.load_state_dict 加载部分预训练参数

比如说,我们没有增删shufflenet的任何部分,只是修改了最后分类器的类别数,那么在对比的过程中,classifier是存在的,classifier部分的预训练参数(类别为1000)就会被加载,依然会报错size mismatch,因为我们将最后的类别数修改为7,和1000不匹配。之前一直使用迁移训练,现记录一下关于model.load_state_dict加载部分预训练参数的学习。我从网上看了很多使用下面的方法,对比自定义模型与预训练模型的键,剔除不同的部分,然后重新加载。

2024-07-08 11:19:43 170

原创 python对文件的基础操作

记录一下python对文件的基础操作。

2024-05-21 16:24:16 176 1

原创 Error loading “D:\Anaconda\envs\pytorch1.8\lib\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll“

OSError: [WinError 126] 找不到指定的模块。我的情况是之前可以用,突然就不能用了,然后回忆了一下在这个期间我为了在VS上调试环境,把cuda里面的一个dll文件的名字后缀改了,即将cublasLt64_11.dll,cublas64_11.dll改成了cublasLt64_10.dll,cublas64_10.dll,我把这两个文件的名字改回去之后就可以正常使用了。经过网上的查询,报错的原因有很多,这里仅记录一下自己的原因,以供参考。

2024-05-11 08:57:16 700 1

原创 docker的命令学习记录

docker run [container ID]/[container name] : 创建并启动一个新容器docker start/stop [container name] :启动或停止一个已经存在的容器docker restart [container name] : 重启容器docker rm [container name] :删除一个或多个容器。

2024-03-18 17:01:51 338

原创 Mlir环境配置以及yolov5模型转换

新手小白记录一下使用算能SE5盒子,在x86主机上配置环境的过程。官方文档为:1. TPU-MLIR简介 — TPU-MLIR 1.6.215 文档 (sophgo.com)

2024-03-16 11:06:28 1038 2

原创 Windows环境下编译onnxruntime学习记录

error MSB4019: 找不到导入的项目“C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\MSBuild\Microsoft\VC\v160\BuildCustomizations\CUDA 11.1.props”注:需要把cuda、cudnn和TensorRT的路径换成自己的。2、从cmd进入上面新建的文件夹 D:\onnxruntime。我的是在:D:\onnxruntime。再重新编译一遍就好了。

2024-01-11 11:38:53 1185 1

原创 关于pyinstaller打包的相关知识记录

(1)打包之后的exe在运行时报错缺少torch,我这里的原因是pyinstaller安装的位置和我的torch不在一起,我是重新新建了一个虚拟环境,用的之前的pyinstaller,在当前环境下再重新安装一下pyinstaller就好了。注:我使用pyinstaller -F xxxx.spec报错。重新生成的exe就没有问题了。

2023-12-26 14:01:25 940 1

原创 使用MobileOne完成图像分类任务

在我自己的图像上的训练结果并不是太好,我使用letterbox做填充的mobilenetv2最后在测试集上的分类精度是99.43%,用s1的分类精度是99.38%,当然对于不用的数据集肯定适用度也不同。最后测试推理速度时发现,推理的速度确实和mobileNetv2的速度也相差无几,和论文里的情况相符。t=N7T8。

2023-12-01 10:12:08 1170 3

原创 CNN分类模型样本不平衡问题

样本不平衡问题指的是某一类的样本数量与其他类别差距特别大,比如有某一类样本数量特别少,或者是某一类的样本数量特别多,这种情况很影响训练阶段的收敛性以及模型在测试集上的泛化性。我遇到的问题是,有一类的样本数量特别少,才几十张图像,与其他类别差距很大,以至于训练好的模型在测试的时候,对这一类别识别的正确率很低。(1)过采样:对数量少的那一个类别的样本进行过采样,即对少样本进行重复采样,以达到平衡不同类别样本的目的。即对样本数量多的类随即删除样本以达到样本数量的平衡,但是针对我的情况,就很不适用。

2023-11-27 10:28:53 453

原创 cuda版本的opencv安装学习参考

安装教程:Cmake报错解决:1、 安装过程中下载超时: 2、报错:CMake Warning at cmake/OpenCVGenSetupVars.cmake:54 (message):directory。

2023-11-27 09:02:44 457

原创 记录ubuntu20.04安装显卡驱动时遇到的问题

1、是了进入tty2,按住按键alt+f2键可进入,但是有的时候按着没反映,我就直接强制关机重启的,进入tty2界面之后,进行显卡的卸载,之后重启可以进入界面化,但是有个问题就是,进入界面化之后,还得重新安装显卡,这里我尝试重新安装显卡,或者是换个版本安装,但是结果都是重启的时候黑屏。然后它就好了,就很莫名其妙,没搞明白。就记录一下,以防之后还出现这种情况。2、修改grub文件(这个方法需要recovery模型先进入界面化 )

2023-11-20 09:38:48 318 1

原创 c++对图像的预处理:

最后得到的float型直接送入到mobilenet模型里面进行推理。

2023-11-09 14:21:16 82 1

原创 WSL+CUDA11.1

用于记录自己的学习过程: 参考文章:【精选】通过WSL2搭建Pytorch1.10+CUDA11.4+NVIDIA Driver深度学习框架_pytorch 11.4-CSDN博客WSL2 安装 CUDA(Win11)_wsl2安装cudnn-CSDN博客CUDA官网:CUDA Toolkit Archive | NVIDIA Developer 选择11.1.0,然后逐步按照下面的顺序选择版本: 根据上面给出的安装命令,逐行进行安装。安装完成之后,测试 nvcc -V显示版本信息: 此时报错的话,不

2023-11-09 13:53:36 51 1

原创 LNK2019 无法解析的外部符号 “public: void __cdecl cv::Mat::copyTo(class cv::debug_build_guard::_OutputArray co

copyTo@Mat@cv@@QEBAXAEBV_OutputArray@debug_build_guard@2@@Z),函数 "void __cdecl preprocess_img(class cv::Mat &,int,int,float * const)" (?preprocess_img@@YAXAEAVMat@cv@@HHQEAM@Z) 中引用了该符号 resize D:\visual studio\Project\resize\resize\letterbox.obj 1。

2023-10-21 10:17:53 709 1

原创 Windows11系统在进行深度模型训练时,报错RuntimeError: CUDA out of memory.

小白记录学习:在windows11系统下训练深度模型的时候,遇到RuntimeError: CUDA out of memory.的情况,可通过设置虚拟的方式解决(当然不是唯一的方法且也不是总有效,原理不太懂,仅记录一下设置的方法以防后面忘记)5、如果初始设定是自动管理的话,需要把哪个勾去掉,然后点击到D盘,再点击自定义大小,需改初始大小和最大值,这个可以随意设置,修改好了之后点击设置,最后确定,返回的每一层都需要点击确定即可。1、首先打开我的电脑的属性,找到高级系统设置。4、找到虚拟内存,点击更改。

2023-10-20 11:06:25 797 2

原创 TensorRT was linked against cudnn 8.6.0 but loaded cudnn 8.0.5

记录学习:在onnx转tensorRT的时候报错TensorRT was linked against cudnn 8.6.0 but loaded cudnn 8.0.5。找到cublasLt64_11.dll将其重命名为cublasLt64_10.dll。接下来就转化成功了!

2023-10-19 10:43:29 541 1

原创 多输出.pth模型转onnx模型,并测试

仅记录一下自己的学习过程,这次的实验,pth模型一共有四个输出,现在需要将其转换成onnx模型。

2023-10-18 10:27:03 241 1

原创 RuntimeError: Exporting the operator repeat_interleave to ONNX opset version 11 is not supported.

在将pth转onnx过程中,报错:RuntimeError: Exporting the operator repeat_interleave to ONNX opset version 9 is not supported.由于repeat_interleave函数是将张量中的元素按照某一维度复制n次,比如torch.repeat_interleave(x, k, dim=2), 因此可以用repeat和view函数代替进行实现。(如有问题,请指正!

2023-10-09 09:41:34 294

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除