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