将图片数据集由TIF转为JPG

将.tif数据集转为.jpg图片

import cv2
import os
import copy
#### 将tif图片转为jpg
def convert_tif2jpg(tif_folder, jpg_folder):  # 输入：tif文件路径，jpg文件路径
    tif_names = os.listdir(tif_folder)
    for i in range(len(tif_names)):
        tif_name = tif_names[i]
        jpg_name = tif_name[:-4]
        jpg_name = jpg_name + '.jpg'
        tif_img = cv2.imread(os.path.join(tif_folder, tif_names[i]))
        print(tif_img.shape)
        print(os.path.join(jpg_folder, jpg_name))
        cv2.imwrite(os.path.join(jpg_folder, jpg_name), tif_img)

convert_tif2jpg('/DRIVE',  '/DRIVEjpg')