Python脚本018：求不同子文件夹的差集

最新推荐文章于 2023-08-06 16:28:00 发布

Coder_L2

最新推荐文章于 2023-08-06 16:28:00 发布

阅读量443

点赞数

分类专栏： python脚本文章标签：深度学习神经网络自动驾驶

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

版权

本文介绍了如何使用Python脚本处理Test数据集，将Query和Gallery两个子集进行对比，找出Gallery中存在而Query中缺失的id。在数据集划分中，由于某些id数量较少，为保持比例，需要从Gallery中抽取相应图片补充到Query。操作包括将相同id的图片归类、计算两者id的差集，并将差集的图片从Gallery转移到Query。

摘要由CSDN通过智能技术生成

背景：在Test数据集中划分Query和Gallery时，会发现有的id的数量比较少，按比例划分的时候Gallery里有而Query里没有，因此就要找出哪些id在Gallery中有而Query没有。所以先把相同id的图片放到同一个文件夹中，对Query和Gallery都遍历一遍，然后发现Query有438个id而Gallery有544个id，所以把差集找出来然后补全到Query，同时从同文件夹的Gallery抽一张图片到Query

import os
import random
import shutil
from shutil import copy2
from IPython import embed


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