实现png与jpg图片的转换。
import cv2
import os
path = r"C:\Users\Shuai\Desktop\DAGM\Class10\data_dataset_voc\SegmentationClassPNG"
newpath = r"C:\Users\Shuai\Desktop\DAGM\Class10\data_dataset_voc\SegmentationClassJPG"
path_list = os.listdir(path)
picture_type = 'jpg'
number = 0 # 统计图片数量
for filename in path_list:
number += 1
portion = os.path.splitext(filename)
print('convert ' + filename + ' to ' + portion[0] + '.' + picture_type)
print(path + "/" + filename)
img = cv2.imread(path + "/" + filename)
print("./" + newpath + "/" + portion[0] + '.' + picture_type)
cv2.imwrite(newpath + "/" + portion[0] + '.' + picture_type, img)
print("共转化了%d张图片" % number)
print('转换完毕,文件存入 ' + newpath + ' 中')
cv2.waitKey(0)
cv2.destroyAllWindows()