原因
直接将png图片扩展名修改成jpg,造成图片失真,需要用OpenCV来对原始图片进行转换。
Python实现
import os
import cv2
if __name__ == '__main__':
dataDir = "F:\YOLO\folder1"
saveDir = "F:\YOLO\folder2"
if not os.path.exists(saveDir):
os.makedirs(saveDir)
for pic in os.listdir(dataDir):
old_path = os.path.join(dataDir, pic)
new_img = cv2.imread(old_path)
# print("new_img", new_img)
new_path = os.path.join(saveDir, pic)
cv2.imwrite(new_path, new_img)
print("图片转换成功!")
不管是绝对路径还是相对路径都不能有中文,否则读取结果为none,特别坑!!
安装cv2
推荐使用清华镜像,否则会很慢……
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple/
整理不易,关注和收藏后拿走!
欢迎专注我的公众号:AdaCoding 和 Github:AdaCoding123