python:从数据集中每个类中挑选出若干样本

可用于制作训练集、gallery等

"""
By Fivethousand
从图片数据集中的每一类都挑选若干张图片,用于制作gallery
"""
import  os
import random
import shutil

dataset_path='F:\python_programs/tasks_from_King\datasets\CASIA-WebFace-part'                 #图片数据集地址
Save_path='F:\python_programs/tasks_from_King\datasets\mytest'                    #保存地址
NUM_OF_IMAGES=10

assert os.path.exists(dataset_path)
os.mkdir(Save_path)
file_lists=os.listdir(dataset_path)
for file in file_lists:
    os.mkdir(os.path.join(Save_path,file))
    file_path=os.path.join(dataset_path,file)
    print("selecting images from:", file_path)
    if len(os.listdir(file_path))<NUM_OF_IMAGES:    # if imgs in this file are insufficient
        continue
    else:
        imgs_list=os.listdir(file_path)
        imgs_selected = random.sample(imgs_list, NUM_OF_IMAGES)  # to select imgs from this file randomly
        for img in imgs_selected:
            img_path=os.path.join(file_path,img)
            shutil.copy(img_path,os.path.join(Save_path,file,img))		#复制
            #如果要变成移动,则用:
            #shutil.move(img_path,os.path.join(Save_path,file,img))	














  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值