全流程零基础自制reid数据集做deepsort

一.如何自制reid数据集

https://blog.csdn.net/DD_PP_JJ/article/details/103374882
darklabel教程标注视频
从这拿reid代码(注意下文件名face_)
注意:他的方法似乎只能处理横屏视频,但我不会修bug/吐舌.
所以我是这么做的:
ffmpeg -i input.mp4 -vf "transpose=1" out.mp4   # 翻转90(所以是,先把竖屏视频翻转成横屏,再使用darklabel标注工具,然后再转换成reid格式)

然后我给个:
首先在aaa(这名字我乱取的)这一文件夹下创建俩空:train,test,
然后把那个创建reid的结果复制到train文件夹下,如果有切割下来的图片小于卷积核大小会报错,所以先使用这个del_small.py 

import os
from PIL import Image

# 目录路径
dir_path = r"C:\Users\Administrator\Desktop\aaa\train"

# 遍历目录中的所有子目录
for root, dirs, files in os.walk(dir_path):
    # 遍历子目录中的所有文件
    for file_name in files:
        file_path = os.path.join(root, file_name)
        # 如果是图片文件，则检查尺寸是否小于128*128像素,小则翻倍
        if file_name.endswith(".jpg") or file_name.endswith(".png"):
            with Image.open(file_path) as img:

                width, height = img.size
                while width < 128 or height < 128:
                    # os.remove(file_path)
                    img = img.resize((width * 2, height * 2))
                    img.save(file_path)
                    width, height = img.size

然后select.py把train里的图拿张放进test

import os
import shutil

# 源目录和目标目录
src_dir = r"C:\Users\Administrator\Desktop\aaa\train"
dst_dir = r"C:\Users\Administrator\Desktop\aaa\test"

# 遍历源目录中的所有子目录
for root, dirs, files in os.walk(src_dir):
    # 遍历子目录中的所有文件
    for file_name in files:
        file_path = os.path.join(root, file_name)
        # 如果是图片文件，则移动到目标目录中同名的子目录中
        if file_name.endswith(".jpg") or file_name.endswith(".png"):
            dir_name = os.path.basename(root)

            folder = os.path.join(dst_dir, dir_name)
            if not os.path.exists(folder):  # 判断是否存在文件夹如果不存在则创建为文件夹
                os.makedirs(folder)

            dst_path = os.path.join(dst_dir, dir_name, file_name)
            shutil.move(file_path, dst_path)
            break  # 只处理一个文件

二.练
https://blog.csdn.net/weixin_39357271/article/details/124760640
yolo部分和deepsort部分大佬都讲清楚了(代码评论区有)
注意一点,如果你的yolo是多分类,记得吧track.py下的--classes ... [0,1,2,...]改成你的分类数
如果不改,就无框了.

https://blog.csdn.net/qq_35832521/article/details/115124521
画线教程,在上一个链接的track.py里修改

三.遇上的一个小问题记录
解决Found no valid file for the classes .ipynb_checkpoints. Supported extensions are: .jpg, .jpeg, .png, .ppm, .bmp, .pgm, .tif, .tiff, .webp ：

rm -rf .ipynb_checkpoints   
find . -name ".ipynb_checkpoints" -exec -a rm -rf {} \;