pytorch
静夜思者
这个作者很懒,什么都没留下…
展开
-
numpy求多维(四维)数组均值
numpy求均值我们知道通过“np.mean(a)”可以求得N为矩阵左右元素的均值,但是有时候我们要求多维数组特定维度的均值我们该怎么办呢?假如我们要求多维数组hsi_target除第二维之外的其他维度的均值。其中hsi_target的维度信息为:(1,31,1392,1300),我们要得到一个长度为31的一维数组。已知:np.mean(a,axis=0)为每列的均值,np.mean(a,axis=1)为求每一行的均值,因此np.mean(a,axis=n)为求第n维的均值,因此我们可以利用嵌套的方法原创 2021-03-11 20:45:33 · 5819 阅读 · 1 评论 -
pytorch函数参数传递不要直接传递模型参数,可直接传递字符串
在主函数中创建模型,其中conv_relu_res_relu_block()函数生成一个网络块,经过resblock()为结构体,将block为主体循环block_num次创建网络,输入通道输出通道数分别为input_channel和output_channel。先建立block,然后再将模型块传递到结构体,该种方法在参数传递的过程中会出现误差。因此不能直接传递block。 block = conv_relu_res_relu_block() print(block) model =原创 2021-03-08 15:16:19 · 1324 阅读 · 2 评论 -
pytorch中numpy矩阵转tensor,矩阵维度变换和维度拓展
np.transpose()将numpy矩阵的纬度按[2,0,1]矩阵交换,np.expand_dims()维度拓展一个维度,位置为axis = 1,维度。一般直接用.copy()进行矩阵复制,不要直接相等。 rgb = np.expand_dims(np.transpose(rgb,[2,0,1]),axis=1).copy()“RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.Flo原创 2021-03-07 09:34:24 · 2351 阅读 · 1 评论 -
Pytorch“ntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict:“
对训练好的模型进行测试,得到测试样本。通过下面的程序将模型参数导入到新建的模型中。由于我们直接用torch.load()存储的模型信息会比较大,因此我们可以只存储参数信息,进行测试时再将参数信息导入到模型中(一定要与保存的模型大小和内容相同)会提高效率。torch.save(model.state_dict(),'hscnn_5layer_dim10_276.pkl')#不直接用torch.save(mode,'hscnn_5layer_dim10_276.pkl')下载保存的模型参数到测试程序:原创 2021-03-04 18:18:23 · 400 阅读 · 1 评论