需求

在使用YOLO时，发现需要的image需要的是jpg格式，而现有的数据集是png格式。
于是需要一个小脚本来进行批量转换

代码

看到有人已经做了相关工作，于是在此基础上稍作修改，完成需求。

import os
from PIL import Image

dirname_read = "D:/Dataset/wangzhe/images/"
dirname_write = "D:/Dataset/wangzhe/images2/"
names = os.listdir(dirname_read)
count = 0
for name in names:
    img = Image.open(dirname_read + name)
    name = name.split(".")
    if name[-1] == "png":
        name[-1] = "jpg"
        name = str.join(".", name)
        # r, g, b, a = img.split()
        # img = Image.merge("RGB", (r, g, b))
        to_save_path = dirname_write + name
        img = img.convert('RGB')
        img.save(to_save_path)
        count += 1
        print(to_save_path, "------conut：", count)
    else:
        continue

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
 

文章来源: zstar.blog.csdn.net，作者：zstar-_，版权归原作者所有，如需转载，请联系作者。

原文链接：zstar.blog.csdn.net/article/details/125832348