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

Coder_L2

于 2021-11-23 15:03:53 发布

阅读量2.5k

点赞数 2

分类专栏： python脚本文章标签：自动驾驶深度学习人工智能

本文链接：https://blog.csdn.net/qq_41876456/article/details/121493251

版权

本文介绍了一个Python脚本，用于将test数据集按照特定比例随机切割为Gallery和Query两个部分。在处理过程中，由于图片数量乘以比例可能非整数导致总数不匹配。提出了两种解决方案：舍弃图片或调整移动文件的逻辑。最终采用了第二种思路来改进代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

目的是将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(

最低0.47元/天解锁文章