sample、 shutil.move 随机抽取文件中图片进行移动（用于从训练集抽取测试集）

问题描述：

将训练集中图片随机移动2份（占比0.1），形成验证集和测试集

完整代码

import shutil
import random
import os

'''
获取所有图片的路径；
移动图片的数目
获取随机移动图片的名称列表
移动图片
'''


# shutil.move(train_path,verify_path)
# random.sample()

def move_file(file_dir, dst_dir):
    path_dir = os.listdir(file_dir)  # 源图像路径
    file_num = len(path_dir)
    rate = 0.1  # 移走图片的比率
    # move_num=int(file_num*rate)
    move_num = 200
    sample = random.sample(path_dir, move_num)  # 随机移动的图片名称
    for name in sample:
        shutil.move(file_dir + name, dst_dir + name)
    return


if __name__ == '__main__':
    # 注意最后有一个斜杠！！！
    train_path = r'G:/00_deeplearning/机器学习/01_code/data/分纤箱外景照/train_image/'  # 原图文件
    aim_path = r'G:/00_deeplearning/机器学习/01_code/data/分纤箱外景照/test_image/'  # 目标文件夹

    move_file(train_path, aim_path)

解决问题思路

1. 获取所有图片路径list

这个时候list是待提取每份图片的名称，并不是完整的路径

存放图片路径：

图片路径列表

2.设置需要移动图片的数目

3.获取随机移动图片的名称列表

4.移动

注意：路径最后