2020.3.20 改原型网络代码遇到的问题

1、numpy库数组属性查看：类型、尺寸、维度、形状

import numpy as np
a=np.array([1,2,3,4],dtype=np.complex128)
print(a) #打印数组a
print(type(a)  #打印a的数据类型
print(a.dtype)  #打印a中元素的数据类型
print(a.size)  #打印数组尺寸，即a中所有元素总数
print(a.shape)  #打印a的数组形状
print(a.ndim)  #打印数组的维度数目

2、代码运行问题：Expected object of scalar type Double but got scalar type Float for argument #2 ‘mat2’

解决方法：将数据类型转为float32类型

data = np.reshape(query_dataSet[n], [1, self.input_shape[0], self.input_shape[1], self.input_shape[2]])
data = data.astype(np.float32)
data = Variable(torch.from_numpy(data))

3、np.ndarray类型转为float类型

data = data.astype(np.float32)

4、cv2.imread() 返回类型

import cv2
img=cv2.imread(path)
print(type(img))

cv2.imread()返回的类型为np.array类型，数组中元素的类型为numpy.uint8