识别非RGB图片并转换为RGB图片

from PIL import Image
import os
path = "C:/Users/user/Desktop/python/Test/image/"
for file_name in os.listdir(path):
  #extension = file_name.split('.')[-1]
  #if extension == 'jpg':
  img =Image.open(path + file_name)
  if img.mode != "RGB" :
    print(file_name)
    img_rgb = img.convert("RGB")
    os.remove(path + file_name)
    img_rgb.save(path + file_name)