Yolov5模型训练过程产生warning：corrupt JPEG restored and saved xxx 解决办法

作者：2023面试高手 | 2024-03-11 14:08:39

踩

corrupt jpeg restored and saved

yolov5进行模型训练的过程中，打印出警告信息：

warning：corrupt JPEG restored and saved xxx
1

事实上这是warning ，并不是error ，不会影响模型的正常训练的代码的运行，但是本质上为了不影响模型最终的训练结果，还是需要把警告消除

代码产生警告原因：
数据集中的图片格式无法读取数据，出现了误差

在算法的训练，进行数据集的数据增广，由于有图片格式不一样的格式，.png格式转换.jpg格式

问题的关键：
.png格式直接转换为.jpg格式，但是直接的转换会造成图片出现失真，所以产生警告。

解决办法:
使用openCV对图像数据进行读取，数据进行重新读取，在写入图片中，相当于进行数据重组

对图片进行Imead读取，在使用iwrite进行图像数据的写入

以下编程的程序代码：

import os
import cv2

train_path = "./train"
train_path1 = "./train1"

val_path 1
2
3
4
5
6

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