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

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

warning:corrupt JPEG restored and saved xxx

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

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

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

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

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

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

以下编程的程序代码:

import os
import cv2

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

val_path 

你可能感兴趣的:(深度学习)