pt文件和转出的onnx文件精度对不上_yolov8 加载onnx格式的模型和加载pt格式的模型,预测同一张图片,预测的置信度不一

作者：寸_铁 | 2024-07-19 07:33:06

踩

yolov8 加载onnx格式的模型和加载pt格式的模型,预测同一张图片,预测的置信度不一

原因：预处理时，图片通道没对齐。。。PIL的image.open和cv2的imread通道顺序不一样。Image.open 读取图片的顺序默认是RGB。cv.imread 读取图片的顺序默认是BGR

解决方法：读取后给图片通道调整顺序即可

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/寸_铁/article/detail/850150