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

最新推荐文章于 2024-03-08 11:01:28 发布

Falcon Leo

最新推荐文章于 2024-03-08 11:01:28 发布

阅读量1.5k

点赞数

文章标签： mot数据集

本文链接：https://blog.csdn.net/weixin_42300418/article/details/112162067

版权

本文介绍了MOT数据集的来源和格式，特别是MOT17-Det的数据处理。详细讲解了如何配置MATLAB环境，包括解决编译问题和使用TDM-GCC编译器。此外，还阐述了motchallenage-devkit官方工具的安装与使用，以及如何进行数据结果的评估，提供了不同数据集的评估接口示例。

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

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[