OPENCV调用PYTORCH的.pth,转onnx,会出现一些差异,

 cv2.dnn.blobFromImage(image, scalefactor=1.0, size, mean, swapRB=True,crop=False,ddepth = CV_32F )

1,转换RGB,因为PIL是RGB格式,而OPENCV是BGR

2,需要归一化,scalefactor=1.0/255

3,最好图片不要进行缩放,因为PIL和OpenCV的缩放函数不一样

4,exp会有一点误差,造成置信度不一样

Logo

免费领 150 小时云算力,进群参与显卡、AI PC 幸运抽奖

更多推荐