Zio_wh-CSDN博客

原创 Anaconda修改镜像源安装PyTorch

Conda 修改镜像源查看目前已有的channel： conda config --show channels添加新的channel：conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://...

2019-12-10 20:27:02 4071

原创 PyTorch读取模型权重：IncompatibleKeys(missing_keys=[], unexpected_keys=[])

IncompatibleKeys(missing_keys=[], unexpected_keys=[])本人在服务器上训练好一个模型后，进行测试时，读入权重文件.pkl(/.t7)时，出现该错误，通过查询源文件torch/nn/modules/module.py中的load_state_dict()，只需将第778行的return _IncompatibleKeys(missing_keys...

2019-08-13 11:19:34 8255

原创 Jupyter Notebook使用parser.parse_args出现错误解决办法

问题：在Jupyter Notebook中使用args传递参数时出现错误：原始代码：args = parser.parse_args()usage: ipykernel_launcher.py [-h] [--x_dim XDIM] [--h_dim HDIM] [--z_dim ZDIM] [--ratio RATIO] [--se...

2019-07-26 09:54:50 7612 14

原创小样本图像识别——关系网络（Relation Network）（目前只对Few-Shot进行研究）

关系网络1.论文参考：《Learning to Compare Relation Network for Few-Shot Learning》论文原文请点击这里参考代码：https://github.com/floodsung/LearningToCompare_FSL/2.问题定义：首先需要了解三个数据集，training set，support set和testing set。s...

2019-07-11 18:11:09 13014 10

原创 pytorch中使用transforms.ToTensor()处理opencv获取的灰度图像问题

pytorch中使用transforms.ToTensor()处理opencv获取的灰度图像问题众所周知，opencv读取的图片的格式为BGR即（height，width，channel），而PIL读取的图片恰恰相反，为（channel，height，width），matplotlib中也是如此。在此插一句：如果要使用matplotlib显示opencv读取的图片，需要将opencv读取的图片...

2019-03-13 16:09:00 8922

原创 MSTAR数据集下载地址及处理方法

首先感谢博主冯子材提供下载的百度网盘地址。在这里为了防止侵权，特此鸣谢，并给出其博客地址。地址中有MSTAR百度网盘下载地址和需要使用的数据介绍，其中百度网盘中有数据转换工具https://blog.csdn.net/taifengzikai/article/details/80600355工具使用因为下载得到的数据是二进制文件，需要使用官方提供的工具进行转换，可将其转换为jpeg。有...

2019-02-22 18:54:32 14922 34

原创 MSI-gtx1060-Tensorflow_gpu-win10-python3.6-cuda9.2 GPU加速安装教程

Tensorflow-GPU加速安装教程需要使用的安装包和版本python3.6：使用Anaconda进行安装，下载地址：https://pan.baidu.com/s/1B6j3bLlV4MwhOXEYn3wo5A提取码：iu5rvs2017：（在此我使用的是community版本）下载地址：https://pan.baidu.com/s/1Lln-I9YM7QT66GAdm1g...

2019-02-20 16:35:55 639 2