自定义博客皮肤VIP专享

*博客头图:

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

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

博客底图:

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

栏目图:

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

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(22)
  • 资源 (7)
  • 收藏
  • 关注

原创 torch.utils.data.dataloader中的collate_fn数据拼接方式那种最快

pytorch自定义数据集加载数据时数据拼接

2022-07-12 16:32:49 353

原创 印度语音数据集下载

(1)Microsoft Speech Corpus (Indian languages)Microsoft Speech Corpus(印度语言)发行版包含泰卢固语、泰米尔语和古吉拉特语的会话和短语语音训练和测试数据(2)Multilingual and code-switching ASR Challenge Dataset - sub-task1(3)Multilingual and code-switching ASR Challenge Dataset - sub-task2...

2021-10-18 12:14:08 305

原创 pytorch中GRU的前向推理过程复现

GRU 学习torch官方资料# -*- coding: utf-8 -*-"""Created on Wed Aug 18 11:11:08 2021@author: yihuang"""import torch.nn as nnimport torchimport torch.nn.functional as Frnn = nn.GRU(10, 10, 1)input = torch.randn(1, 3, 10)output, hn = rnn(input)#复现推理

2021-08-18 16:37:59 594 1

原创 pytorch加载网络权重并冻结部分权重

需求构建一个新的网络,加载一个已经训练好的网络的公共部分。解决import torchfrom UnetModelsCuaseNotPadding_TwoStreamT2 import CRN_Netdef fix_bn(m): classname = m.__class__.__name__ if classname.find('BatchNorm') != -1: m.eval() model = CRN_Net()pretext_model = tor

2021-05-27 14:03:42 2566

转载 git的add、commit、push的详细介绍

简单的代码提交流程git status 查看工作区代码相对于暂存区的差别 git add . 将当前目录下修改的所有代码从工作区添加到暂存区 . 代表当前目录 git commit -m ‘注释’ 将缓存区内容添加到本地仓库 git pull origin master先将远程仓库master中的信息同步到本地仓库master中 git push origin master 将本地版本库推送到远程服务器, origin是远程主机,master表示是远程服务器上的master分支和本地分支重名的

2021-05-10 10:24:11 1344

原创 通过atan实现atan2函数

参考MATLAB函数——atan(x)和atan2(x)的区别在有的情况下只提供了atan的内置函数而没有atan2,先要判断象限再调整。import torchimport mathpi = math.pireal = torch.tensor([ 0.12, -1.1, -0.1, 1])imag = torch.tensor([ 0.22, 1.2, -1.2, -1.2])atan = torch.atan(imag/real)print(atan)atan[1] = ata

2021-04-29 13:49:59 1538 1

原创 PyTorch自定义带学习参数的激活函数(如sigmoid)

有的时候我们需要给损失函数设一个超参数但是又不想设固定阈值想和网络一起自动学习,例如给Sigmoid一个参数alpha进行调节函数如下:import torch.nn as nnimport torchclass LearnableSigmoid(nn.Module): def __init__(self, ): super(LearnableSigmoid, self).__init__() self.weight = torch.nn.Paramete

2021-04-15 17:58:49 4053

原创 Ubuntu网卡配置,节点配置

1.配置一个静态IP使用dhclient分配一个IP地址也可以自己设sudo dhclientifconfig网卡配置sudo vim /etc/network/interfacesauto enp0s3iface enp0s3 inet staticaddress 192.168.127.159netmask 255.255.255.0g...

2020-04-22 21:30:04 697

原创 卷积神经网络的总结

目录1.CNN简介2.CNN本质特性2.1 稀疏连接(稀疏互交)2.2 参数共享2.3小结3.CNN物理机制3.1线性的卷积3.2 离散卷积和相似度3.3小结4.CNN的经典模型及特点4.1经典网络4.2多层CNN卷积的构造参考1.CNN简介卷积神经网络(CNN)是一种前馈神经网络,其特点是每一层的的神经元节点只响应前一层局部范围内的神经...

2019-07-18 16:09:19 2713

原创 pytorch加载预训练网络权重,单GPU加载多GPU,CPU加载GPU权重

保存网络权重:torch.save(G_AB.state_dict(), 'G_AB.pth')单GPU加载权重:G_AB.load_state_dict(torch.load(G_AB.pth))CPU加载单GPU权重:g.load_state_dict(torch.load('./save/G.pth',map_location='cpu')) #load weig...

2019-06-20 15:46:41 3038

原创 赛码网刷题python

偶尔一道编程题锻炼思维1.题目股神http://exercise.acmcoder.com/online/online_judge_ques?ques_id=1664&konwledgeId=134观察规律发现减法的位置的规律,只要找到当前数前面有几个减法,可以求得输出。例如 14 - 3*2=8 (三个减号)代码:#!/usr/bin/env python # ...

2018-10-15 18:51:52 4080

原创 ubuntu16.04安装opencv3+opencv-contrib以及遇到cuda问题安装失败解决

1.opencv3第一步:安装依赖包;sudo apt-get install build-essentiasudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-devsudo apt-get install python-dev pyth...

2018-09-27 10:47:23 1459

原创 python中对fits文件读取和保存

天文图片一般都是以.fits的格式保存的。安装包 astropysudo pip3 install astropy图片的读取:from astropy.io import fitsimport oshud = fits.open('/home/yellow/1.fits')img = hud[0].data图片的保存path = '/home/yellow/'...

2018-08-28 15:31:53 10467 3

原创 高效卷积算法

总结下卷积加速的三种实现方法:方案一:卷积等效于使用傅里叶变换将输入与核都转换到复频域,做一个点乘运算,再用逆变换变回到实域,这的确比离散的卷积更快。方案二:当卷积核是可以分离的可以拆成一列乘一行的情况(可以用SVD验证一个卷积核是否可拆),将列与输入进行卷积后再把结果与行进行卷积,这种情况做卷积是最快的但是它只是针对特定的卷积核。这里提供大家一个链接里面专门讲解了和对比了这种方法...

2018-04-15 11:13:50 5861

原创 远程与ssh相关的命令 and 远程终端运行程序可以显示图片

远程登入 带端号的IP (注 p为小写)ssh [email protected] -p 35000数据上传,下载(注:不用登入远程端,在本地端就好了)上传数据 (注:没有端口号的不用打 -P 35000 下同)scp -P 35000 /home/yellow/1.txt [email protected]:/home/lab30201//home/ye...

2018-03-14 16:38:52 738

原创 GPU共享内存小结---pycuda

先来点题外话,由于导师的要求使用GPU加速但是又没有系统的学习就开始了使用pycuda进行GPU的编程,变用边学。在优化程序的时候shared memory是必用的,所以理解共享内存是个什么东西才可以做到自己对数据是如何跑的有数。 先看看这张GPU的存储结构图(偷的图,哈哈^_^): 在图中我们关心的重点是,每一个block都对应的一个shared memory,线程块之可以相互交流通...

2017-10-26 21:48:48 10709 4

原创 ubunu下安装nvidia显卡驱动,卸载cuda8安装cuda10.1,cuDnn

虽然网上有许多教程,本帖记录下,方便自己以后查找,同时也方便他人。安装显卡驱动 第一步在英伟达官网下载相对应的显卡驱动 https://www.geforce.cn/drivers第二步开始安装 禁用nouveau sudo vim /etc/modprobe.d/blacklist.conf 进入 最后一行 添加blacklist nouveau更新内核 sudo up...

2017-10-16 13:04:32 395

原创 复数变量传递(超简单)----pycuda

有时我们需要在pycuda里传递复数进行处理,C 语言中加一个头文件就好了,超简单哦! 比起我之前用的那个简单多了,还可以调用cuComplex.h里的函数,如取实部,虚部等等。 看看代码小例子吧!import pycuda.driver as cudaimport pycuda.autoinitfrom pycuda.compiler import SourceModuleimport

2017-09-29 11:48:39 1078

原创 FFT快速卷积GPU加速----pycuda

二维卷积定理的实现1.读取一幅图f(x,y)大小A*B ,一个卷积核h(x,y)大小为C*D。 2.对f,h,进行填充,填充成Q*P,P>=A+C-1,Q>=B+D-1。 3.分别对f,h做傅里叶变换(FFT2) 4.G=F*H(注意是乘法不是点乘) 5.对G做IFFT2 6.对5中的结果进行剪切得到原大小的图片。CPU下numpy实现直接看代码,其中填充是手动填充的。#-*- codin

2017-09-15 10:40:41 2790

原创 复数类型矩阵相乘GPU加速--pycuda

开发环境Ubuntu16.04 ,python3复数类型矩阵相乘GPU加速–pycuda我在用卷积定理做卷积时遇到的一个问题就两矩阵做完FFT之后都是为复数怎么用pycuda做矩阵相乘,在给GPU传递参数的时候总是有问题。

2017-09-10 19:50:46 3100

原创 在GPU下做fft和Ifft----pycuda

1.调用keikna库keikna库中有fft,所以为了减小任务量我就调用keikan的fft的库来完成。http://blog.sina.com.cn/s/blog_4513dde60102vstu.html这个帖子里详细的介绍了keikna的一些东西。2.介绍FFT和IFFT的实现对于二维的傅里叶变换的实现,冈萨雷斯《数字图像处理》有详细的介绍。通过对这两个

2017-09-10 12:55:32 5042

原创 2D卷积GPU加速---pycuda

第一次写帖子,也想长期在这里写帖子,一起交流学习。直接给代码import numpyimport pycuda.autoinitimport pycuda.driver as cudafrom pycuda.compiler import SourceModuleTILE_W = 16TILE_H = 16KERNEL_RADIUS = 1KERNEL_W = 2 * K...

2017-09-05 10:11:02 1753 1

Python卷积用矩阵乘法实现

卷积用矩阵乘法实现,这么做非常适合GPU加速,有喜欢的朋友可以看看,顺便给小的打赏点积分,谢谢!

2018-04-15

MATLAB代码卷积运算转化为乘法

MATLAB代码卷积运算转化为乘法,本来就不可能但是这真的是可以的,看看吧

2018-04-15

Split bregman 算法MATLAB源代码

人家一篇《bregman alteration》论文都卖10分,我代码就要5分,不是原创就是分享,我就想赚点积分。

2018-03-14

算法导论_中文版_第三版

PDF 中文高清。

2017-10-02

空空如也

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

TA关注的人

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