numpy 指定 astype 为 float32 结果却是 float64 的原因
在图像预处理函数中,你使用 image = image.astype(np.float32) 将图像数组转换为 float32 类型。然而,打印结果显示为 float64。
这是因为后续的操作 image = (image – mean) / std 中,mean 和 std 都是 float64 类型。浮点运算遵循精度最高的输入类型,因此结果也变为 float64。
示例代码
image = np.array([1, 2, 3]) image = image.astype(np.float32) mean = np.array([0.5, 1.0, 1.5]).astype(np.float64) std = np.array([0.2, 0.3, 0.4]).astype(np.float64) # 结果为 float64 result = (image - mean) / std print(result.dtype) # 输出:float64