python脚本实现将某一路径下的UCMD分类数据集格式转换

python脚本实现将某一路径下的UCMD分类数据集格式转换（tif–>jpeg/其他）

from PIL import Image
import os

def transimg(path):
    #path:imgdataset_dir
    for filename in os.listdir(path):
        img_path = path + '/' + filename
        str = img_path.rsplit(".", 1)
        output_img_path = str[0] + ".jpeg"
        print(output_img_path)
        im = Image.open(img_path)
        rgb_im = im.convert('RGB')
        rgb_im.save(output_img_path)
        #yi qu zhiqian de
        os.remove(img_path)
            
     

if __name__ == '__main__':
    class_dict = {'agricultural': 0,
                  'airplane': 1,
                  'baseballdiamond': 2,
                  'beach': 3,
                  'buildings': 4,
                  'chaparral': 5,
                  'denseresidential': 6,
                  'forest': 7,
                  'freeway': 8,
                  'golfcourse': 9,
                  'harbor': 10,
                  'intersection': 11,
                  'mediumresidential': 12,
                  'mobilehomepark': 13,
                  'overpass': 14,
                  'parkinglot': 15,
                  'river': 16,
                  'runway': 17,
                  'sparseresidential': 18,
                  'storagetanks': 19,
                  'tenniscourt': 20}
    for item in class_dict.keys():
       path='/xxx/xx/xx...'+item
       print(transimg(path))