Generate JDE dot train

Dot train format for JDE

• Generate train Format file
• Filter by the format of the dataset

from glob2 import glob
import os

images = r"C:\Users\lpf\Desktop\MOT\Code-MOT\Towards-Realtime-MOT-master\data\KITTI_Det\images"
dot_train = r"C:\Users\lpf\Desktop\MOT\Code-MOT\Towards-Realtime-MOT-master\data\kitti_det.train"

path_labels = r"C:\Users\lpf\Desktop\MOT\Code-MOT\Towards-Realtime-MOT-master\data\KITTI_Det\labels_with_ids"


def write_dot_train(path_dot_train, img_path_list):
    with open(path_dot_train, "w") as f:
        for path in img_path_list:
            f.writelines("{}\n".format(path))

#  This method is detection Special for dataset conversion 
def generate_dot_train(path_labels):
    include_object = []
    include_img = []
    labels_txt = glob(path_labels + "/*.txt")
    for i in labels_txt:
        file_size = os.path.getsize(i)
        if file_size > 0:
            include_object.append(i)
            # print("File size is :", file_size, "bytes")
            # print(i)
            a = os.path.split(i)[-1]
            temp = a[:-4]
            temp = temp + ".png"
            # temp = os.path.join(images, temp)
            temp = "KITTI_Det/images/{}".format(temp)
            include_img.append(temp)
            # break
    return include_img
        # pass


include_image = generate_dot_train(path_labels=path_labels)
write_dot_train(dot_train,include_image)

# print(len(labels_txt))
# print(len(include_object))

#  This method is  tracking special 
def generate_kitti_train():
    #  Note that the generated here is based on image Benchmarking train file , Include all pictures , And usually after screening lebels_with_ids No way. , Will be less 
    #  Modification is the use of  lebels_with_ids  Benchmarking , And then txt  Use replace  Methods to png
    path = os.walk(images)
    for root, directories, files in path:
        for _dir in directories:
            line = "/KITTI_Det/images/{}/".format(_dir)
            _dir = os.path.join(r"E:\8_DataSet\KITTI_tracking\image_02", _dir)
            # dir_list.append(_dir)
            txt_list = os.listdir(_dir)
            with open("kitti-img.train", 'a') as f:
                for item in txt_list:
                    line1 = line + item
                    f.writelines(line1)
                    f.writelines("\n")