python 随机划分图片数据集

一个随机划分图片数据集的方法。

任务描述：我的所有图片保存在同一个文件夹里，需要随机划分为训练集和测试集。

处理过程：读取文件列表，将列表打乱，截取列表一部分

import os
import random
import shutil


def get_imlist(path):
    return [os.path.join(path, f) for f in os.listdir(path) if f.endswith('.bmp')]


def getData(src_path):
    dest_dir = './test'  # 这个文件夹需要提前建好
    img_list = get_imlist(src_path)
    random.shuffle(img_list)
    le = int(len(img_list) * 0.8)  # 这个可以修改划分比例
    for f in img_list[le:]:
        shutil.move(f, dest_dir)


getData('./test_data2')