这次我在做项目时,数据集有些事png图像,训练时出现下面的信息
虽然这个并不影响最终的结果,但是看的不舒服
png格式的图像通道不能直接更改为jpg格式,需要改变通道
from PIL import Image
import os
# 设置需要转换的图片目录
img_dir = r""
# 遍历目录下所有文件
for filename in os.listdir(img_dir):
# 判断文件是否为png格式
if filename.endswith(".png"):
# 构造新的文件名,将后缀改为jpg
new_filename = os.path.splitext(filename)[0] + ".jpg"
# 打开图片
img = Image.open(os.path.join(img_dir, filename))
# 如果图片是RGBA模式,转换为RGB模式
if img.mode == "RGBA":
img = img.convert("RGB")
# 如果图片是P模式,转换为RGB模式
if img.mode == "P":
img = img.convert("RGB")
# 保存为新的jpg格式图片
img.save(os.path.join(img_dir, new_filename))
# 关闭图片
img.close()
# 删除原来的png格式图片
os.remove(os.path.join(img_dir, filename))
只改变后缀,不改变前缀