笔记的代码参考博主【只会GAN的小朋友】http://t.csdn.cn/uj9YP的博客。
1:在MOT16数据集下创建一个 MOT_yolo.py 文件。
2:将转换代码复制到MOT_yolo.py文件中。
# MOT数据集gt.txt标签转换yolo标签
'''
创建以下四个目录,用于存放图片和标签
images/train
images/val
labels/train
labels/val
'''
import os
import shutil
import numpy as np
import configparser
if not os.path.exists('images'):
os.makedirs('images/train')
os.makedirs('images/val')
if not os.path.exists('labels'):
os.makedirs('labels/train')
os.makedirs('labels/val')
def convert(imgWidth, imgHeight, left, top, width, height):
x = (left + width / 2.0) / imgWidth
y = (top + height / 2.0) / imgHeight
w