deepsort按照自己的类别分类

deepsort按照自己的数据集类别分类

[参考文章]

(https://blog.csdn.net/weixin_53711236/article/details/123762215)

按照该博主的代码先进行检测目标裁剪，再进行数据分类。本人在该博主数据分类代码的基础上进行了改动，使分类后的文件夹全部是一类图片，在后续model.py中num_class改动中可以写自己数据集的类别
代码如下：

import os
from PIL import Image
from shutil import copyfile, copytree, rmtree, move

PATH_DATASET = './crops'  # 需要处理的文件夹
PATH_NEW_DATASET = './test'  # 处理后的文件夹
PATH_ALL_IMAGES = PATH_NEW_DATASET + '/all_images'
PATH_TRAIN = PATH_NEW_DATASET + '/train'
PATH_TEST = PATH_NEW_DATASET + '/test'


# 定义创建目录函数
def mymkdir(path):
    path = path.strip()  # 去除首位空格
    path = path.rstrip("\\")  # 去除尾部 \ 符号
    isExists = os.path.exists(path)  # 判断路径是否存在
    if not isExists:
        os.makedirs(path)  # 如果不存在则创建目录
        print(path + ' 创建成功')
        return True
    else:
        # 如果目录存在则不创建，并提示目录已存在
        print(path + ' 目录已存在')
        return False


class BatchRename():
    '''
    批量重命名文件夹中的图片文件
    '''

    def __init__(self):
        self.path = PATH_DATASET  # 表示需要命名处理的文件夹

    # 修改图像尺寸
    def