mot数据集_基于MATLAB的MOT官方评估工具

MOT数据集

数据集地址：MOT Challenge

数据集格式：MOT Challenge，MOT17数据集格式_bug++-CSDN博客_mot数据集格式

本实验是基于mmdetection框架faster rcnn，数据集格式为将MOT17-Det转化成VOC格式进行训练，对于训练集的结果利用以下脚本转化成标准格式

数据集处理及训练参照以下几篇文章：

HUST小菜鸡：将MOT17-Det数据集转成VOC格式​zhuanlan.zhihu.com

HUST小菜鸡：通过HSV转换的方式实现图片数据增强​zhuanlan.zhihu.com

HUST小菜鸡：使用mmdetection测试以及生成检测结果​zhuanlan.zhihu.com

import pickle
# import cv2
import numpy as np
import os

fp_pkl = open('faster_voc_train1.pkl','rb')
inf = pickle.load(fp_pkl)
length = len(inf)

# print(inf[:5])
print(length)
fp_txt = open('train1.txt','r')
train_list = fp_txt.readlines()
print(train_list[100])

output_dir = 'D:/Pycharm/MOT-DET/result'
if not os.path.exists(output_dir):
    os.makedirs(output_dir)
# for i in range (len(test_list)):

fp2 = open(output_dir+'/MOT17-02.txt','w')
fp4 = open(output_dir+'/MOT17-04.txt','w')
fp5 = open(output_dir+'/MOT17-05.txt','w')
fp9 = open(output_dir+'/MOT17-09.txt','w')
fp10 = open(output_dir+'/MOT17-10.txt','w')
fp11 = open(output_dir+'/MOT17-11.txt','w')
fp13 = open(output_dir+'/MOT17-13.txt','w')

for i in range(length):
    pic_info = train_list[i]
    file_fp = pic_info[