图片重命名或者将.jpg后缀与.png后缀的修改

作者介绍

张伟伟，男，西安工程大学电子信息学院，2019级硕士研究生，张宏伟人工智能课题组。
研究方向：机器视觉与人工智能。
微信公众号：可随时查阅，搜索—张二牛的笔记，回复“python‘’获取整理好的专栏。
电子邮件：zhangweiweicpp@163.com

• 课题组CSDN官方账号，欢迎一键三连！.

已测试，可用

import cv2
import os
#MAIN_PATH为数据集的文件路径，注意最后末尾的反斜杠 / 需要加上,test_dir 为需要保存的图片路径。
MAIN_PATH="D:/DCGAN/smalldata/"
test_dir = "./test_dir/"
if not os.path.exists(test_dir): os.makedirs(test_dir)

def get_imgs():
    files = os.listdir(MAIN_PATH)
    imgs = []
    for file in files:
        # print(MAIN_PATH + file)
        imgs.append(cv2.imread(MAIN_PATH + file))
    print(" get images successfully")
    return imgs

#得到训练数据的图片
real_img = get_imgs()
print("数据集的图片数目：",len(real_img))

for i in range(len(real_img)):
    img = real_img[i]
    cv2.imwrite(test_dir + str(i) + ".png", img, )
    
print("完成")

博主辛苦整理，不吝一键三连啊！！！