关于#python#的问题：用cv2实现图片的分类存储

Question

用python cv2编程：

多个不同文件夹，每个文件夹里都存了一张彩图jpg文件，一张红外png图片，图片不同，要求：

1、新建一个彩图文件夹，一个红外文件夹；

2、彩色文件夹中按编号1no ,2no规则存入所有彩色图，红外文件夹中存入所有红外图，编号要和彩色文件夹中编号一致。

怎么编程实现从多个文件夹中分别读取其中的图片，按图片的后缀格式分类存储到不同的文件夹中。

那位帮帮忙解答一下。感谢感谢。

Answer 1

import os
os.mkdir('彩色文件夹路径')
os.mkdir('红外文件夹路径')
c_num = 0
r_num = 0
for dir in os.listdir('存放多个文件夹的总文件夹路径'):
    for file in dir:
        if file.endswith('jpg'):
            os.rename('存放多个文件夹的总文件夹路径' + dir + file, '彩色文件夹路径' + str(c_num) + 'no.jpg')
            c_num += 1
        else:
            os.rename('存放多个文件夹的总文件夹路径' + dir + file, '红外文件夹路径' + str(r_num) + 'no.jpg')
            r_num += 1