pytorch推理过程中出现此错误,可能是输入端的numpy数组中值的数据类型是double类型
在numpy数组中,np.double等于np.float64即double类型,而np.float32为float类型,因此在生成数组时,令dtype = np.float32即可解决该问题。
例如:img = np.zeros((1,3,640,640), dtype = np.float32)
pytorch推理过程中出现此错误,可能是输入端的numpy数组中值的数据类型是double类型
在numpy数组中,np.double等于np.float64即double类型,而np.float32为float类型,因此在生成数组时,令dtype = np.float32即可解决该问题。
例如:img = np.zeros((1,3,640,640), dtype = np.float32)