- 博客(21)
- 资源 (1)
- 收藏
- 关注
原创 pytorch学习系列(10):断点续训
训练中难免遇到停电、宕机等情况。如果没有设置断点续训,那么就得从头开始,非常浪费时间和能源。要实现断点续训,首先要保存之前训练的模型,除此之外,还有优化器的状态。这是因为目前很多优化器的参数是会随着训练而不断变化的,具体可见这篇博客:pytorch学习系列(4):常用优化算法首先设置两个参数:start_epoch:续训时开始训练的epoch数resume:是否实施断点续训start_e...
2020-04-12 11:49:09 3273 9
原创 压缩感知图像重建质量与JPEG2000(kakadu)的批量比较
本脚本适合于压缩感知的研究者,将自己的算法与JPEG2000进行比较,批量得出结果。
2019-11-25 21:28:40 764
原创 图像PSNR和SSIM批量比较,适用于图像重建
该脚本适合于图像压缩、压缩感知、超分辨率、去噪、去雾和去雨等问题的研究,批量将处理后的图像和原图进行比较,得到PSNR和SSIM值,保存到汇总表格。
2019-11-25 20:30:14 1711 4
转载 pytorch学习系列(8):统计模型的参数和计算量
https://mp.weixin.qq.com/s/EXuFXbPBIbzTyi0fUjvvPw
2019-11-25 15:39:02 815
转载 pytorch学习系列(7):代码调试工具torchsnooper
https://mp.weixin.qq.com/s/PVIWWIqbZuEe4lwVMDYA5Q
2019-11-25 15:37:02 249
原创 pytorch学习系列(5):单通道灰度图的保存(大坑)
pytorch的torchvision无法保存真正的灰度图像,只能产生三个相同通道的伪灰度图像,本文就是解决这个问题的
2019-11-25 15:16:17 4737
原创 十进制和二进制的转换
二进制转十进制(1011)2=1∗23+1∗21+1∗20=8+2+1=11(1011)_{2}=1*2^{3}+1*2^{1}+1*2^{0}=8+2+1=11(1011)2=1∗23+1∗21+1∗20=8+2+1=11对于小数,把整数部分和小数部分分开计算即可十进制转二进制整数:除2取余,逆序排列11:11/2=5余111/2=5 余111/2=5余15/2=2余15/2=2...
2019-10-24 20:28:51 559
原创 pytorch学习系列(5):模型部分参数使用(迁移学习)
有两种方式:1.Net.load_state_dict(torch.load(model_path),strict=False)使用strict参数,如果为True,表明预训练模型的层和自己定义的网络结构层严格对应相等(比如层名和维度),这里选择为False,则不完全对等,会自动舍去多余的层和其参数。2.pretrained_dict=torch.load(model_path)mo...
2019-10-11 15:31:23 1242
原创 适合于图像重建问题的归一化层:GENERALIZED NORMALIZATION TRANSFORMATION(GDN)
一种适合于图像生成和恢复的归一化层及其pytorch实现
2019-09-04 22:38:02 6303 2
原创 kakadu——JPEG2000图像压缩软件的安装和使用
JPEG2000是图像压缩界最常用的压缩算法,而kakadu是jpeg2000最高效的实现,也是最常用的。本文介绍在ubuntu系统下使用kakadu的过程
2019-08-05 22:38:34 5038 18
原创 cuDNN error: CUDNN_STATUS_EXECUTION_FAILED和cuda的卸载和安装
cuDNN error: CUDNN_STATUS_EXECUTION_FAILED这种错误在网上的资料很少,有人说是因为python,pytorch,cuda等的版本不适配导致的,有人将python换成3.6版本的就解决了,但有的没有。在我这里,原因出在显卡和cuda的不适配上,我用的RTX2080ti,本来装的是cuda9,就出现了上述bug,经过多次探索,将cuda换成10后,问题解决,估...
2019-04-24 16:38:39 10590
原创 安装win7虚拟机和orCAD软件
安装windows7虚拟机及PSPICE 10.5软件PSPICE 10.5在win8和win10上不能正常运行,因为安装完成后会出现软件未响应,CPU占满100%的奇怪情况,所以只能在win7及以下的系统上运行。因此如果自己的电脑上不是win7系统,则可以选择安装win7虚拟机来运行PSPICE 10.5系列软件。预备工作安装Oracle VM VirtualBox;下载windows...
2019-04-15 21:51:55 920
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人