python os.listdir将文件中所有图进行单独图片保存在对应文件夹中

eye123456789

已于 2022-03-17 10:46:14 修改

阅读量2.4k

点赞数 2

文章标签： python 开发语言计算机视觉

于 2022-03-17 10:43:16 首次发布

本文链接：https://blog.csdn.net/eye123456789/article/details/123544535

版权

#将一个文件中所有的图片进行另保存，并且对应创建文件夹保存对应图
import os
import os.path
import cv2
read_path='C:/Users/qhq/Downloads/imgs'
#save_path='C:/Users/qhq/Downloads/imgs1/'
#save_path='//nas/LargeSave/gaoqing_images/httpswww.pexels.comzh-tw/'
save_path='C:/Users/qhq/Downloads/imgs1/'
def read_directory(directory_name):
    count = 0
    for filename in os.listdir(directory_name):#获取指定文件中的所有文件文件和子文件组成的列表
        img = cv2.imread(directory_name + "/" + filename)
        print(filename)
        if os.path.isdir(save_path + str(count)):
            pass
        else:
            os.mkdir(save_path + str(count))
        document_path = save_path + str(count)
        pic_path = document_path + '/' + str(count) + '.jpg'
        #print(pic_path)

        #####保存图片#########
        cv2.imwrite(pic_path, img)
        count += 1

read_directory(read_path)