简介
此代码可以随机移动大文件夹中的十分之一图片到小文件夹中,可以用于在深度学习收集的大量数据中选出其中用来训练的部分数据
代码:
import os
import random
from shutil import move
# 设置源文件夹和目标文件夹路径
source_folder = '/home/build/下载/PIC/pic1500'
destination_folder = '/home/build/下载/PIC/cccc'
# 确保目标文件夹存在
if not os.path.exists(destination_folder):
os.makedirs(destination_folder)
# 获取所有图片文件的列表
images = [f for f in os.listdir(source_folder) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
# 计算要移动的图片数量(大约是十分之一,但至少一张)
num_images_to_move = max(1, len(images) // 10)
# 随机选择要移动的图片
images_to_move = random.sample(images, num_images_to_move)
# 移动图片
for image in images_to_move:
src = os.path.join(source_folder, image)
dst = os.path.join(destination_folder, image)
move(src, dst)
print(f'Moved: {
image}')
print('Done moving images.')