Linux下随机（通过复制文件）划分训练，验证，测试集（5:3:2）

from shutil import copyfile
from os import walk
import random

soc_cover_path = 'xxx' #源文件路径
soc_stego_path = 'xxx'

dst_cover_path = 'xxx'#目标文件路径
dst_stego_path = 'xxx'

random.seed(123456)
for (dirpath, dirnames, filenames) in walk(soc_cover_path):
    count = 0
    indexs = []
    for i in range(len(filenames)):
        index = random.randrange(0, len(filenames))
        while index in indexs:
            index = random.randrange(0, len(filenames))
        indexs.append(index)
        count += 1
        name = filenames[index]

        if count <= 5000:
            copyfile(soc_cover_path + name, dst_cover_path + 'train/' + name)
            copyfile(soc_stego_path + name, dst_stego_path + 'train/' + name)
        elif count <= 8000:
            copyfile(soc_cover_path + name, dst_cover_path + 'validation/' + name)
            copyfile(soc_stego_path + name, dst_stego_path + 'validation/' + name)
        else:
            copyfile(soc_cover_path + name, dst_cover_path + 'test/' + name)
            copyfile(soc_stego_path + name, dst_stego_path + 'test/' + name)