实现代码如下:

import os
import glob
from PIL import Image

root_dir = './image_net_mini'
dirs = os.listdir(root_dir)
for path in dirs:
    JPEG_list = glob.glob(root_dir + '/' + path + '/*.jpg')
    for jpeg in JPEG_list:
        img = Image.open(jpeg)
        if img.mode != 'RGB':
            img = img.convert('RGB')
            print(jpeg)
            os.remove(jpeg)
            img.save(jpeg)

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐