- 博客(6)
- 收藏
- 关注
原创 pytorch中squeeze()和unsqueeze()函数介绍;MATLAB中的squeeze()
squeeze()是来减少一个维度的unsqueeze()是来增加一个维度的pytroch中代码:mask=np.load(’/home/yy/cd‘)mask = torch.tensor(mask) #pytroch中要输入tensor,mask原来为numpy,转化成tensorprint(mask.shape,'初始的mask维度') mask = mask.unsqueeze(-1) #在最后增添一维 print(mask.shape,'增加一个维度以后'
2021-01-29 00:01:59
1957
原创 峰值信噪比PSNR~均方差MSE matlab实现
峰值信噪比PSNR~均方差MSE matlab实现峰值信噪比:PSNR用来评价一幅图像压缩后和原图像相比质量的好坏。PSNR越高,压缩后失真越小,效果越好。均方差用MSE表示,均方差值越小越好。均方差是各数据偏离真实值的距离平方和的平均数。简单的介绍一下,下面是代码实现;MSE:%out1是我们的输入%out2是我们的输出方法一:在malab有直接的内置函数mse,简单粗暴!实例:MSE=mse(out1-out2);方法二:我们根据MSE公式来计算:[h w]=size(out1);%读
2021-01-28 23:23:47
3102
1
原创 linux下为项目创建一个新的python虚拟环境env
创建一个新的虚拟环境首先你需要安装一个Anaconda要不然其余的免谈查看一下自己是否有cuda终端输入cuda,如图所示。开始创建终端输入conda create -n <env_name> <package_name><env_name>—是你虚拟环境的名字<package_name>–你需要的python版本如conda create -n hq python=3.6注意!用conda装就是python=3.6,用pip就是 pyth
2020-11-27 21:02:39
1944
2
原创 Linux20.0.4 CUDA(11.0)版本卸载
在命令行输入cd /usr/local/cuda/bin./cuda-uninstaller然后出现的页面中将所有选项全选(在这个选项上按空格键即可选择)然后Done就可以了出现Successfully uninstalled 就成功了
2020-11-26 22:31:12
1042
原创 ## 初入坑小白linux20.0.4下安装cuda11.0和cudnn
初入坑小白linux20.0.4下安装cuda11.0和cudnn在安装cuda和cudnn时首先你要理清一个思路,就是先确定自己linux的版本号,你要运行程序里pytorch或是tensflow的版本号。因为cuda和cudnn及pytorch是要对应起来的否则,后续调用不起来相当于白装(这一点灰常重要!!!)本人就是血淋淋例子至于怎么查呢,百度一下你就知道了。理清楚在动手也不迟!PS:可能用到的命令查看自己的linux版本号: cat /etc/issue查询CUDA版本输入命令:
2020-11-26 22:01:39
4790
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅