人脸识别0-09：insightFace-半监督系统搭建(4)-数据集分割

以下链接是个人关于insightFace所有见解，如有错误欢迎大家指出，我会第一时间纠正，如有兴趣可以加微信：17575010159 相互讨论技术。 人脸识别0-01：insightFace目录:https：//blog.csdn.net/weixin_43013761/article/details/99646731： 这是本人项目的源码：https://github.com/944284742/1.FaceRecognition 其中script目录下的文件为本人编写，主要用于适应自己的项目，可以查看该目录下的redeme文件。

数据集分割

根据前面的博客，我们已经对数据做了人脸提取，分类，以及数据清洗，现在我们需要对我们的数据集进行分割，一般有两种情况，情况1：只有训练集和验证集，情况2：存在训练集和验证集以及测试集。下面是我编写的脚本，两种情况都包揽在其中，script\data_preproce.py：

from __future__ import print_function
import os
import sys

curr_path = os.path.abspath(os.path.dirname(__file__))
sys.path.append(os.path.join(curr_path, "../python"))
import mxnet as mx
import random
import argparse
import cv2
import time
import traceback
import shutil

try:
    import multiprocessing
except ImportError:
    multiprocessing = None

def dirs_rename(args):

    newdir_idx = args.stdir_idx
    for old_path, dirs, files in os.walk(args.root, followlinks=True):
        files.sort()
        dirs.sort()
        if old_path == args.root:
            continue

        if (args.delete_mixnums != 0)   and (len(files)