![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 63
张嘉烘
这个作者很懒,什么都没留下…
展开
-
whl本地安装mamba-ssm
项目中用到MambaIR时,需要pip安装 causal_conv1d和 mamba-ssm两个包及其依赖。但是在安装过程中可能会出现building wheel卡着不动后error…的whl文件,并pip install本地安装。原创 2024-07-01 14:20:02 · 177 阅读 · 1 评论 -
虚拟环境导致的段错误
事情开始是我想把代码复制到另一台服务器跑,所以代码是一模一样的,但是在配置环境之后运行训练代码却出现了这个错误。网上关于这个问题的解决方法也是千奇百怪,但是基本都解决不了我的问题。经过几天的摸索,才发现原来是配置的环境出了问题。所以改为如下命令安装,再重新运行,无报错。关于pytorch的安装,我参考的是这篇文章。,大家也可以自行查找其他文章,大差不差。段错误(核心已转储)后来发现,应该装的是。原创 2024-06-05 10:54:31 · 346 阅读 · 0 评论 -
pytorch迁移学习训练图像分类
适用于训练数据集与预训练模型相差大时,可以选择微调训练所有层,此时只使用预训练模型的部分权重和特征,例如原始模型为imageNet,而训练数据为医疗相关。总体而言,迁移学习能够利用已有的知识和经验,加速模型的训练过程,提高模型的性能。对于微调全连接层的选择一,测试集准确率为 72.078%而所有权重随机的选择三测试集准确率为 43.228%4,下载数据集压缩包(下载之后需要解压数据集)调用不同迁移学习得到的模型对比测试集准确率。2,安装Pytorch。原创 2023-09-17 17:06:53 · 1064 阅读 · 0 评论 -
pytorch生成CAM热力图-单张图像
6,创建 test_img 文件夹,并下载测试图像到该文件夹。4,下载中文字体文件(用于显示和打印汉字文字)5,下载 ImageNet 1000类别信息。7,下载安装 torchcam。3,安装 mmcv-full。2,安装 Pytorch。原创 2023-09-15 12:12:29 · 2340 阅读 · 0 评论 -
Transformer关于Self-Attention及Multi-Head Attention
对于 RNN 模型记忆长度有限且无法并行化,只有计算完 t(i) 时刻后的数据才能计算 t(i+1) 时刻的数据。Transformer 则有效的弥补了这个问题。转载文章中作者对 Self-Attention 进行了详细的讲解,在此基础上引申出 Multi-Head Attention。转载 2023-08-11 15:36:50 · 43 阅读 · 0 评论 -
自定义MyDataSet获取数据及对应label
实例化数据集需要用到 DataSet 类,我们可以自定义来实现对数据集的处理。原创 2023-08-03 21:43:23 · 845 阅读 · 1 评论 -
深度学习——划分自定义数据集
以人脸表情数据集raf_db为例,初始目录如下:需要经过处理后返回定义。原创 2023-08-02 21:51:15 · 1119 阅读 · 1 评论 -
ResNet网络结构入门
当网络堆叠到一定深度时,深层网络的效果可能比浅层网络效果差。原创 2023-07-10 17:14:00 · 251 阅读 · 0 评论 -
pytorch搭建VGG网络
VGG 网络的创新点:通过堆叠多个小卷积核来替代大尺度卷积核,可以减少训练参数,同时能保证相同的感受野。例如,可以通过堆叠两个 3×3 的卷积核替代 5x5 的卷积核,堆叠三个 3×3 的卷积核替代 7x7 的卷积核。原创 2023-06-26 10:42:01 · 1021 阅读 · 0 评论 -
pytorch搭建AlexNet网络实现花分类
使用Dropout的方式在网络正向传播过程中随机失活一部分神经元,以减少过拟合。原创 2023-06-17 17:55:52 · 2890 阅读 · 0 评论 -
pytorch搭建LeNet网络实现图像分类器
model.py。原创 2023-06-15 10:58:32 · 1339 阅读 · 0 评论