数据预处理-三光源图片分类

对仅有三张光场图片的数据集进行归类，方便后续脚本进行批量处理。

import os
import shutil


def read_List():
    list = os.listdir(os.getcwd())
    with open("list.txt", 'w') as file:
        file.writelines(line + '\n' for line in list if line.endswith('.png'))


def classify():
    with open('list.txt') as file:
        list = file.readlines()
    data_num = len(list) / 3
    for i in range(int(data_num)):
        # 创建目标目录
        file_destination = './' + str(i)
        if not os.path.exists(file_destination):
            os.mkdir(str(i))
        remain_files = os.listdir(os.getcwd())
        remain_files = [x for x in remain_files if x.endswith('.png')]
        mark = remain_files[0][0:5]
        for j in remain_files:
            if mark in j:
                shutil.move(j, file_destination)