Python脚本015:将test数据集切割成Gallery和Query

目的是将Test数据集中分割成比例的Gallery和Query
然后要求是Gallery和Query都需要同id下的图片
我写了两段代码
1.先把同id的先放在文件夹中,由于有2542个id就有2542个文件夹
2.在同id的文件夹中按比例随机分配给Gallery和Query

Input:
在这里插入图片描述
Output:
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述


1.把同id的先放在一个文件夹中


# 导入对路径操作的库
import sys
import os
from IPython import embed
import shutil
import glob

def mkDir(path,id):
    isExists = os.path.exists(os.path.join(path, id))
    if not isExists:						#判断如果文件不存在,则创建
        os.makedirs(os.path.join(path, id))


if __name__ == '__main__':

    gallery_path = '/home/jy/xl/workstation/Datasets/Car/Nighttime_Vehicle_ReID/gallery'  # 设置创建后文件夹存放的位置
    query_path = '/home/jy/xl/workstation/Datasets/Car/Nighttime_Vehicle_ReID/query' # 设置创建后文件夹存放的位置
    test_path = '/home/jy/xl/workstation/Datasets/Car/Nighttime_Vehicle_ReID/test'  # 图片存放的位置
    test_temp_path = '/home/jy/xl/workstation/Datasets/Car/Nighttime_Vehicle_ReID/test_temp'

    img_names = os.listdir(test_path)

    for img_name in img_names:
        str_list = img_name.split("_")
        # str_list = img_path.split(
  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值